PyPI page
Home page
Author:
Marco Simoni
License:
MIT
Summary:
A decoder-only Transformer built from scratch in JAX and Flax NNX
Latest version:
0.3.15
Required dependencies:
flax
|
jax
|
jaxlib
|
msgpack
|
optax
|
pyyaml
|
tokenizers
|
tqdm
Optional dependencies:
build
|
dantinox
|
datasets
|
huggingface-hub
|
matplotlib
|
mkdocs-glightbox
|
mkdocs-material
|
mkdocs-minify-plugin
|
mkdocstrings
|
mypy
|
numpy
|
pandas
|
pytest
|
pytest-cov
|
ruff
|
scipy
|
twine
|
types-pyyaml
Downloads last day:
103
Downloads last week:
1,643
Downloads last month:
1,643