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.6
Required dependencies:
click
|
colorlog
|
datasets
|
hydra-core
|
liger-kernel
|
lm-eval
|
mlflow
|
omegaconf
|
psutil
|
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
|
mypy
|
nv-grouped-gemm
|
pandas-stubs
|
pre-commit
|
pytest
|
pytest-cov
|
pytest-xdist
|
ruff
|
seaborn
|
transformer-engine
|
types-psutil
|
types-tqdm
Downloads last day:
1
Downloads last week:
6
Downloads last month:
138