A collection of pretrained models in Flax
Flax Models The goal of this project is to make current deep learning models more easily available for the awesome Jax/Flax ecosystem. Models Example Notebooks to play with on Colab Installation You will need Python 3.7 or later. For GPU usage, follow the Jax installation with CUDA. Then install: > pip install –upgrade git+https://github.com/matthias-wright/flaxmodels.git For CPU-only you can skip step 1. Documentation The documentation for the models can be found here. Checkpoints The checkpoints are taken from the repositories that […]
Read more