shyamsn97 / hyper-nn
Easy Hypernetworks in Pytorch and Jax
☆98Updated 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
- Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper☆80Updated 3 years ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆158Updated last year
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- Sequence Modeling with Structured State Spaces☆63Updated 2 years ago
- Unofficial Implementation of "Denoising Diffusion Probabilistic Models" in PyTorch(Lightning)☆89Updated 4 years ago
- Repository for the "Gotta Go Fast When Generating Data with Score-Based Models" paper☆104Updated 3 years ago
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆122Updated last year
- LoRA for arbitrary JAX models and functions☆135Updated last year
- Implementation of Recurrent Interface Network (RIN), for highly efficient generation of images and video without cascading networks, in P…☆201Updated last year
- Artificial Kuramoto Oscillatory Neurons☆66Updated 3 weeks ago
- ☆164Updated 2 years ago
- Score-Based Generative Modeling with Critically-Damped Langevin Diffusion☆197Updated last year
- ☆33Updated 2 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
- ☆60Updated 3 years ago
- Contrastive Language-Image Pretraining☆142Updated 2 years ago
- Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.☆249Updated last year
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆108Updated 2 years ago
- ☆49Updated 2 years ago
- Replicating and dissecting the git-re-basin project in one-click-replication Colabs☆36Updated 2 years ago
- FID computation in Jax/Flax.☆27Updated 7 months ago
- Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch☆99Updated 2 years ago
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆52Updated 4 months ago
- Running Jax in PyTorch Lightning☆89Updated 2 months ago
- Differentiable Algorithms and Algorithmic Supervision.☆113Updated last year
- A minimalist implementation of score-based diffusion model☆122Updated 3 years ago
- ☆88Updated 2 years ago
- Gradient-based constrained optimization for JAX☆29Updated 2 years ago
- Code for ICLR 2021 Paper, "Anytime Sampling for Autoregressive Models via Ordered Autoencoding"☆26Updated last year
- ICML 2022: Learning Iterative Reasoning through Energy Minimization☆45Updated 2 years ago