├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── examples └── finetune_bert.py ├── mezo_jax ├── __init__.py ├── mezo_jax.py └── utils.py ├── requirements.txt └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvm23/mezo-jax/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvm23/mezo-jax/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvm23/mezo-jax/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvm23/mezo-jax/HEAD/README.md -------------------------------------------------------------------------------- /examples/finetune_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvm23/mezo-jax/HEAD/examples/finetune_bert.py -------------------------------------------------------------------------------- /mezo_jax/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvm23/mezo-jax/HEAD/mezo_jax/__init__.py -------------------------------------------------------------------------------- /mezo_jax/mezo_jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvm23/mezo-jax/HEAD/mezo_jax/mezo_jax.py -------------------------------------------------------------------------------- /mezo_jax/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvm23/mezo-jax/HEAD/mezo_jax/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | jax 2 | -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vvvm23/mezo-jax/HEAD/setup.py --------------------------------------------------------------------------------