abarcel / haikumodelsLinks
Pretrained models for Jax/Haiku; MobileNet, ResNet, VGG, Xception.
☆24Updated 3 years ago
Alternatives and similar repositories for haikumodels
Users that are interested in haikumodels are comparing it to the libraries listed below
Sorting:
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆112Updated 3 years ago
- ☆33Updated 2 years ago
- AdaCat☆49Updated 3 years ago
- Very deep VAEs in JAX/Flax☆46Updated 4 years ago
- 👩 Pytorch and Jax code for the Madam optimiser.☆51Updated 4 years ago
- Image augmentation library for Jax☆39Updated last year
- Easy Hypernetworks in Pytorch and Jax☆103Updated 2 years ago
- ☆153Updated 5 years ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆179Updated 2 weeks ago
- Gaussian-Bernoulli Restricted Boltzmann Machines☆104Updated 2 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 3 weeks ago
- Meta Optimal Transport☆103Updated 2 years ago
- Normalizing Flows in Jax☆107Updated 4 years ago
- 🧀 Pytorch code for the Fromage optimiser.☆125Updated last year
- Tensorflow 2.0 implementation of Fourier Feature Mapping Networks.☆43Updated 5 years ago
- Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.☆256Updated 4 months ago
- Repository for the paper "Simple and Effective VAE Training with Calibrated Decoders"☆29Updated 4 years ago
- Fast Discounted Cumulative Sums in PyTorch☆96Updated 3 years ago
- A discrete sequential VAE☆40Updated 5 years ago
- ☆105Updated last year
- Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper☆81Updated 3 years ago
- ☆50Updated 4 years ago
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆125Updated last year
- Official code for the Stochastic Polyak step-size optimizer☆139Updated last year
- Jupyter Notebook corresponding to 'Going with the Flow: An Introduction to Normalizing Flows'☆27Updated 4 years ago
- Lightweight Cluster/Cloud VM Job Management 🚀☆42Updated 11 months ago
- ☆36Updated 2 years ago
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 5 years ago
- ☆68Updated last year
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆258Updated last year