abarcel / haikumodelsLinks
Pretrained models for Jax/Haiku; MobileNet, ResNet, VGG, Xception.
☆23Updated 3 years ago
Alternatives and similar repositories for haikumodels
Users that are interested in haikumodels are comparing it to the libraries listed below
Sorting:
- AdaCat☆49Updated 2 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆112Updated 3 years ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆179Updated last month
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- 👩 Pytorch and Jax code for the Madam optimiser.☆51Updated 4 years ago
- Jupyter Notebook corresponding to 'Going with the Flow: An Introduction to Normalizing Flows'☆27Updated 4 years ago
- Normalizing Flows in Jax☆107Updated 4 years ago
- ☆104Updated last year
- PyTorch implementation of Continuously Indexed Flows paper, with many baseline normalising flows☆31Updated 3 years ago