PyPI page
Home page
Author:
aetherBERT contributors
License:
Apache-2.0
Summary:
A DeepSpeed-like training library for Google TPUs, built on JAX/Flax.
Latest version:
0.1.0
Required dependencies:
datasets
|
flax
|
jax
|
numpy
|
optax
|
orbax-checkpoint
|
pyyaml
|
transformers
|
wandb
Optional dependencies:
aetherlm
|
jupyter
|
mteb
|
nmn
|
pytest
|
ruff
|
scikit-learn
|
scipy
|
torch
Downloads last day:
0
Downloads last week:
11
Downloads last month:
40