PyPI page
Home page
Author:
None
License:
MIT
Summary:
A clean, modular framework for training large language models with modern PyTorch features
Latest version:
0.2.0
Required dependencies:
click
|
colorlog
|
datasets
|
hydra-core
|
liger-kernel
|
lm-eval
|
mlflow
|
omegaconf
|
pydantic
|
pyyaml
|
rich
|
scipy
|
tiktoken
|
torch
|
torchdata
|
torcheval
|
tqdm
|
transformers
|
wandb
|
zstandard
Optional dependencies:
black
|
griffe-fieldz
|
ipython
|
isort
|
jupyter
|
matplotlib
|
mkdocs
|
mkdocs-gen-files
|
mkdocs-literate-nav
|
mkdocs-material
|
mkdocstrings
|
pre-commit
|
pytest
|
pytest-cov
|
pytest-xdist
|
ruff
|
seaborn
Downloads last day:
2
Downloads last week:
37
Downloads last month:
78