PyPI page
Home page
Author:
David Hall, Jason Wang, Ahmed Ahmed, Ivan Zhou, Will Held, Virginia Adams
Summary:
Scalable Training for Foundation Models with Named Tensors and JAX
Latest version:
0.2.41.dev202607090907
Required dependencies:
async-lru
|
braceexpand
|
chex
|
dataclasses-json
|
datasets
|
deepdiff
|
draccus
|
einops
|
equinox
|
filelock
|
flash-attn-4
|
fsspec
|
gcsfs
|
humanfriendly
|
immutabledict
|
jax
|
jax-triton
|
jaxtyping
|
jinja2
|
jmp
|
lenses
|
marin-fray
|
marin-haliax
|
marin-iris
|
marin-rigging
|
marin-zephyr
|
nvidia-cublas
|
nvidia-cutlass-dsl
|
nvidia-nccl-cu13
|
optax
|
protobuf
|
pyarrow
|
pydantic
|
pytimeparse
|
safetensors
|
tblib
|
tensorstore
|
tokenizers
|
tqdm-loggable
|
transformers
|
triton
|
wandb
|
zstandard
Optional dependencies:
fastapi
|
google-api-python-client
|
google-auth
|
google-cloud-storage
|
google-cloud-storage-transfer
|
jax
|
jaxlib
|
libtpu
|
openai
|
peft
|
tensorboard
|
tensorboardx
|
tokamax
|
torch
|
uvicorn
|
xprof
Downloads last day:
144
Downloads last week:
1,279
Downloads last month:
5,150