LICENSE
README.md
setup.py
src/nextgenjax/__init__.py
src/nextgenjax/activations.py
src/nextgenjax/custom_layers.py
src/nextgenjax/layers.py
src/nextgenjax/model.py
src/nextgenjax/optimizers.py
src/nextgenjax/parallelism_example.py
src/nextgenjax/train.py
src/nextgenjax/transformer_models.py
src/nextgenjax/vocab_parallel_embedding.py
src/nextgenjax.egg-info/PKG-INFO
src/nextgenjax.egg-info/SOURCES.txt
src/nextgenjax.egg-info/dependency_links.txt
src/nextgenjax.egg-info/requires.txt
src/nextgenjax.egg-info/top_level.txt