PyPI page
Home page
Author:
License:
Summary:
Scalable Training for Foundation Models with Named Tensors and JAX
Latest version:
1.1
Required dependencies:
braceexpand
|
dataclasses-json
|
datasets
|
draccus
|
equinox
|
fsspec
|
gcsfs
|
haliax
|
humanfriendly
|
jaxtyping
|
jmp
|
matplotlib
|
optax
|
pyarrow
|
pydantic
|
pytimeparse
|
ray
|
requests
|
rich
|
safetensors
|
tblib
|
tensorstore
|
transformers
|
wandb
|
zstandard
Downloads last day:
1,463
Downloads last week:
6,518
Downloads last month:
22,483