shyamsn97 / hyper-nn
Easy Hypernetworks in Pytorch and Jax
☆100Updated 2 years ago
Alternatives and similar repositories for hyper-nn:
Users that are interested in hyper-nn are comparing it to the libraries listed below
- LoRA for arbitrary JAX models and functions☆136Updated last year
- Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.☆251Updated last month
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆109Updated 2 years ago
- Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper☆80Updated 3 years ago
- ☆33Updated 2 years ago
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- Official implementation of Transformer Neural Processes☆75Updated 2 years ago
- AdaCat☆49Updated 2 years ago
- FID computation in Jax/Flax.☆27Updated 9 months ago
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆127Updated 2 years ago
- Parameter-Free Optimizers for Pytorch☆126Updated last year
- Official Pytorch and JAX implementation of "Efficient-VDVAE: Less is more"☆194Updated 2 years ago
- Use Jax functions in Pytorch☆239Updated last year
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆123Updated last year
- A collection of meta-learning algorithms in Jax☆23Updated 2 years ago
- Fast Discounted Cumulative Sums in PyTorch☆95Updated 3 years ago
- Unofficial JAX implementations of deep learning research papers☆156Updated 2 years ago
- Sequence Modeling with Structured State Spaces☆63Updated 2 years ago
- Code for the paper: Rotating Features for Object Discovery☆51Updated 8 months ago
- Repository for the "Gotta Go Fast When Generating Data with Score-Based Models" paper☆104Updated 3 years ago
- Implementation of PSGD optimizer in JAX☆33Updated 4 months ago
- ☆36Updated last year
- ☆62Updated 3 years ago
- ☆51Updated 2 years ago
- Gaussian-Bernoulli Restricted Boltzmann Machines☆104Updated 2 years ago
- Contrastive Language-Image Pretraining☆142Updated 2 years ago
- NF-Layers for constructing neural functionals.☆84Updated last year
- The 2D discrete wavelet transform for JAX☆43Updated 2 years ago
- A new play-and-plug method of controlling an existing generative model with conditioning attributes and their compositions.☆73Updated 3 years ago
- Pytorch-like dataloaders for JAX.☆80Updated last week