PyPI page
Home page
Author:
None
Summary:
Architecture experimentation and training infrastructure.
Latest version:
0.2.1
Required dependencies:
click
|
datasets
|
flax
|
jsonlines
|
loguru
|
numpy
|
omegaconf
|
orbax
|
pydantic
|
python-dotenv
|
rich
|
seaborn
|
tiktoken
|
torchax
|
wandb
Optional dependencies:
aiofiles
|
bcrypt
|
coverage
|
coveralls
|
fastapi
|
itsdangerous
|
jax
|
jinja2
|
mkdocs
|
mkdocs-gen-files
|
mkdocs-literate-nav
|
mkdocs-material
|
mkdocs-section-index
|
mkdocstrings
|
mypy
|
pre-commit
|
pymdown-extensions
|
pytest
|
pytest-cov
|
python-multipart
|
pyyaml
|
ruff
|
sse-starlette
|
tokenizers
|
torch
|
torchax
|
transformers
|
uvicorn
|
watchdog
|
wikipedia
Downloads last day:
28
Downloads last week:
242
Downloads last month:
288