PyPI page
Home page
Author:
None
Summary:
MaxText is a simple, performant and scalable Jax LLM!
Latest version:
0.2.1
Required dependencies:
cffi
|
hf-xet
|
jax-cuda12-pjrt
|
jax-cuda12-plugin
|
libtpu
|
nvidia-cublas-cu12
|
nvidia-cuda-cupti-cu12
|
nvidia-cuda-nvcc-cu12
|
nvidia-cuda-nvrtc-cu12
|
nvidia-cuda-runtime-cu12
|
nvidia-cudnn-cu12
|
nvidia-cufft-cu12
|
nvidia-cusolver-cu12
|
nvidia-cusparse-cu12
|
nvidia-nccl-cu12
|
nvidia-nvjitlink-cu12
|
nvidia-nvshmem-cu12
|
pycparser
|
pywin32
|
ruamel-yaml-clib
Optional dependencies:
absl-py
|
aiofiles
|
aiohappyeyeballs
|
aiohttp
|
aiohttp-cors
|
aiosignal
|
annotated-doc
|
annotated-types
|
anthropic
|
antlr4-python3-runtime
|
anyio
|
aqtp
|
argcomplete
|
array-record
|
astor
|
astroid
|
asttokens
|
astunparse
|
attrs
|
auditwheel
|
black
|
blake3
|
blobfile
|
boto3
|
botocore
|
build
|
cachetools
|
cbor2
|
certifi
|
cffi
|
cfgv
|
charset-normalizer
|
cheroot
|
chex
|
click
|
cloud-accelerator-diagnostics
|
cloud-tpu-diagnostics
|
cloudpickle
|
clu
|
cmake
|
colorama
|
colorful
|
comm
|
compressed-tensors
|
contourpy
|
coverage
|
cryptography
|
cycler
|
dacite
|
dataclasses-json
|
datasets
|
debugpy
|
decorator
|
depyf
|
dill
|
diskcache
|
distlib
|
distro
|
dm-tree
|
dnspython
|
docker
|
docstring-parser
|
drjax
|
durationpy
|
editdistance
|
einops
|
einshape
|
email-validator
|
entrypoints
|
etils
|
evaluate
|
execnet
|
executing
|
fastapi
|
fastapi-cli
|
fastapi-cloud-cli
|
fastar
|
fastjsonschema
|
filelock
|
flatbuffers
|
flax
|
fonttools
|
frozenlist
|
fsspec
|
gast
|
gcsfs
|
gguf
|
google-api-core
|
google-api-python-client
|
google-auth
|
google-auth-httplib2
|
google-auth-oauthlib
|
google-cloud
|
google-cloud-aiplatform
|
google-cloud-appengine-logging
|
google-cloud-audit-log
|
google-cloud-bigquery
|
google-cloud-common
|
google-cloud-core
|
google-cloud-filestore
|
google-cloud-logging
|
google-cloud-mldiagnostics
|
google-cloud-monitoring
|
google-cloud-resource-manager
|
google-cloud-storage
|
google-cloud-storage-control
|
google-crc32c
|
google-genai
|
google-metrax
|
google-pasta
|
google-resumable-media
|
google-tunix
|
googleapis-common-protos
|
grain
|
grpc-google-iam-v1
|
grpcio
|
grpcio-reflection
|
grpcio-status
|
gspread
|
gviz-api
|
h11
|
h5py
|
hf-transfer
|
hf-xet
|
httpcore
|
httplib2
|
httptools
|
httpx
|
httpx-sse
|
huggingface-hub
|
humanize
|
hypothesis
|
identify
|
idna
|
ijson
|
immutabledict
|
importlab
|
importlib-metadata
|
importlib-resources
|
iniconfig
|
interegular
|
ipykernel
|
ipython
|
ipython-pygments-lexers
|
ipywidgets
|
isort
|
jaraco-classes
|
jaraco-context
|
jaraco-functools
|
jax
|
jaxlib
|
jaxtyping
|
jedi
|
jeepney
|
jinja2
|
jiter
|
jmespath
|
joblib
|
jsonlines
|
jsonschema
|
jsonschema-specifications
|
jupyter-client
|
jupyter-core
|
jupyterlab-widgets
|
kagglehub
|
keras
|
keyring
|
keyrings-google-artifactregistry-auth
|
kiwisolver
|
kubernetes
|
lark
|
latex2sympy2-extended
|
libclang
|
libcst
|
libtpu
|
llguidance
|
llvmlite
|
lm-format-enforcer
|
loguru
|
lxml
|
markdown
|
markdown-it-py
|
markupsafe
|
marshmallow
|
math-verify
|
matplotlib
|
matplotlib-inline
|
mccabe
|
mcp
|
mdurl
|
mistral-common
|
ml-collections
|
ml-dtypes
|
ml-goodput-measurement
|
model-hosting-container-standards
|
more-itertools
|
mpmath
|
msgpack
|
msgspec
|
multidict
|
multiprocess
|
mypy-extensions
|
myst-nb
|
myst-parser
|
namex
|
nbclient
|
nbformat
|
nest-asyncio
|
networkx
|
ninja
|
nixl
|
nltk
|
nodeenv
|
numba
|
numpy
|
numpy-typing-compat
|
nvidia-cublas-cu12
|
nvidia-cuda-cupti-cu12
|
nvidia-cuda-nvrtc-cu12
|
nvidia-cuda-runtime-cu12
|
nvidia-cudnn-cu12
|
nvidia-cufft-cu12
|
nvidia-cufile-cu12
|
nvidia-curand-cu12
|
nvidia-cusolver-cu12
|
nvidia-cusparse-cu12
|
nvidia-cusparselt-cu12
|
nvidia-nccl-cu12
|
nvidia-nvjitlink-cu12
|
nvidia-nvshmem-cu12
|
nvidia-nvtx-cu12
|
oauthlib
|
omegaconf
|
openai
|
openai-harmony
|
opencensus
|
opencensus-context
|
opencv-python-headless
|
opentelemetry-api
|
opentelemetry-exporter-otlp
|
opentelemetry-exporter-otlp-proto-common
|
opentelemetry-exporter-otlp-proto-grpc
|
opentelemetry-exporter-otlp-proto-http
|
opentelemetry-exporter-prometheus
|
opentelemetry-proto
|
opentelemetry-sdk
|
opentelemetry-semantic-conventions
|
opentelemetry-semantic-conventions-ai
|
opt-einsum
|
optax
|
optree
|
optype
|
orbax-checkpoint
|
orbax-export
|
outlines-core
|
packaging
|
pandas
|
papermill
|
parameterized
|
parso
|
partial-json-parser
|
pathspec
|
pathwaysutils
|
perfetto
|
pexpect
|
pillow
|
platformdirs
|
pluggy
|
portpicker
|
pre-commit
|
prometheus-client
|
prometheus-fastapi-instrumentator
|
promise
|
prompt-toolkit
|
propcache
|
proto-plus
|
protobuf
|
psutil
|
ptyprocess
|
pure-eval
|
py-cpuinfo
|
py-spy
|
pyarrow
|
pyasn1
|
pyasn1-modules
|
pybase64
|
pycnite
|
pycountry
|
pycparser
|
pycryptodomex
|
pydantic
|
pydantic-core
|
pydantic-extra-types
|
pydantic-settings
|
pydot
|
pyelftools
|
pyglove
|
pygments
|
pyink
|
pyjwt
|
pylatexenc
|
pylint
|
pyparsing
|
pyproject-hooks
|
pytest
|
pytest-mock
|
pytest-xdist
|
python-dateutil
|
python-dotenv
|
python-json-logger
|
python-multipart
|
pytype
|
pytz
|
pyyaml
|
pyzmq
|
qwix
|
ray
|
referencing
|
regex
|
requests
|
requests-oauthlib
|
rich
|
rich-toolkit
|
rignore
|
rpds-py
|
rsa
|
ruamel-yaml
|
runai-model-streamer
|
runai-model-streamer-gcs
|
runai-model-streamer-s3
|
s3transfer
|
safetensors
|
scipy
|
scipy-stubs
|
secretstorage
|
sentencepiece
|
sentry-sdk
|
seqio
|
setproctitle
|
setuptools
|
setuptools-scm
|
shapely
|
shellingham
|
shortuuid
|
simple-parsing
|
simplejson
|
six
|
smart-open
|
sniffio
|
sortedcontainers
|
sphinx
|
sphinx-book-theme
|
sphinx-copybutton
|
sphinx-design
|
sse-starlette
|
stack-data
|
starlette
|
supervisor
|
sympy
|
tabulate
|
tenacity
|
tensorboard
|
tensorboard-data-server
|
tensorboard-plugin-profile
|
tensorboardx
|
tensorflow
|
tensorflow-datasets
|
tensorflow-metadata
|
tensorflow-text
|
tensorstore
|
termcolor
|
tiktoken
|
tokamax
|
tokenizers
|
toml
|
tomlkit
|
toolz
|
torch
|
torchax
|
torchvision
|
tornado
|
tpu-info
|
tqdm
|
traitlets
|
transformer-engine
|
transformer-engine-cu12
|
transformer-engine-jax
|
transformers
|
treescope
|
triton
|
typeguard
|
typer
|
typing-extensions
|
typing-inspect
|
typing-inspection
|
tzdata
|
uritemplate
|
urllib3
|
uv
|
uvicorn
|
uvloop
|
virtualenv
|
wadler-lindig
|
watchfiles
|
wcwidth
|
websocket-client
|
websockets
|
werkzeug
|
wheel
|
widgetsnbextension
|
wrapt
|
xgrammar
|
xpk
|
xprof
|
xxhash
|
yapf
|
yarl
|
zipp
|
zstandard
Downloads last day:
76
Downloads last week:
494
Downloads last month:
3,905