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:
- 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
- Image augmentation library for Jaxβ39Updated last year
- AdaCatβ49Updated 2 years ago
- Very deep VAEs in JAX/Flaxβ46Updated 4 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).β112Updated 3 years ago
- β36Updated 2 years ago
- Normalizing Flows in Jaxβ108Updated 4 years ago
- Repository for the paper "Simple and Effective VAE Training with Calibrated Decoders"β29Updated 3 years ago
- Efficiently Composable Data Augmentation on the GPU with Jaxβ37Updated last month
- β68Updated last year
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation preconditionβ¦β177Updated 2 weeks ago
- JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"β43Updated 4 years ago
- Lightweight Cluster/Cloud VM Job Management πβ41Updated 9 months ago
- β33Updated 2 years ago
- A small library for creating and manipulating custom JAX Pytree classesβ56Updated 2 years ago
- Easy Hypernetworks in Pytorch and Jaxβ101Updated 2 years ago
- MineRL Navigate Video Datasetβ12Updated 4 years ago
- β192Updated 2 months ago
- Clockwork VAEs in JAX/Flaxβ32Updated 3 years ago
- Generalised UDRLβ37Updated 3 years ago
- β153Updated 5 years ago
- Fast Discounted Cumulative Sums in PyTorchβ96Updated 3 years ago
- Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwoβ¦β70Updated 3 weeks ago
- β104Updated last year
- β49Updated 4 years ago
- β36Updated last year
- Texture mapping with variational auto-encodersβ40Updated 3 years ago
- Gaussian-Bernoulli Restricted Boltzmann Machinesβ104Updated 2 years ago
- Accelerated replay buffers in JAXβ41Updated 2 years ago