.gitignore
LICENSE
README.md
THIRD_PARTY_NOTICES.md
pyproject.toml
setup.py
.github/workflows/build-and-test.yml
.github/workflows/release.yml
src/README.md
src/jax_hf/__init__.py
src/jax_hf/jax_modules.py
src/jax_hf/main.py
src/jax_hf/mixing.py
src/jax_hf/utils.py
src/jax_hf/wrappers.py
src/jax_hf.egg-info/PKG-INFO
src/jax_hf.egg-info/SOURCES.txt
src/jax_hf.egg-info/dependency_links.txt
src/jax_hf.egg-info/requires.txt
src/jax_hf.egg-info/top_level.txt