shyamsn97 / hyper-nnLinks
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
Sorting:
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- Sequence Modeling with Structured State Spaces☆64Updated 2 years ago
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆124Updated last year
- Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper☆81Updated 3 years ago
- Implementation of Mega, the Single-head Attention with Multi-headed EMA architecture that currently holds SOTA on Long Range Arena☆204Updated last year
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆157Updated last year
- Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch☆100Updated 2 years ago
- Implementation of Discrete Key / Value Bottleneck, in Pytorch☆88Updated last year
- LoRA for arbitrary JAX models and functions☆136Updated last year
- Replicating and dissecting the git-re-basin project in one-click-replication Colabs☆36Updated 2 years ago
- NF-Layers for constructing neural functionals.☆85Updated last year
- FID computation in Jax/Flax.☆27Updated 10 months ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆110Updated 3 years ago
- Gaussian-Bernoulli Restricted Boltzmann Machines☆104Updated 2 years ago
- Repository for the "Gotta Go Fast When Generating Data with Score-Based Models" paper☆104Updated 3 years ago
- Unofficial JAX implementations of deep learning research papers☆156Updated 2 years ago
- The Official PyTorch Implementation of "VAEBM: A Symbiosis between Variational Autoencoders and Energy-based Models" (ICLR 2021 spotlight…☆56Updated 2 years ago
- ☆228Updated 3 months ago
- A minimalist implementation of score-based diffusion model☆127Updated 3 years ago
- Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.☆254Updated 2 months ago
- Implementation of Recurrent Interface Network (RIN), for highly efficient generation of images and video without cascading networks, in P…☆204Updated last year
- Use Jax functions in Pytorch☆242Updated last year
- Running Jax in PyTorch Lightning☆102Updated 5 months ago
- ☆52Updated 2 years ago
- ICML 2022: Learning Iterative Reasoning through Energy Minimization☆46Updated 2 years ago
- ☆33Updated 2 years ago
- Implementation of PSGD optimizer in JAX☆33Updated 5 months ago
- Flow-matching algorithms in JAX☆92Updated 9 months ago
- ☆163Updated 2 years ago
- Pytorch-like dataloaders for JAX.☆83Updated last week