Alternatives and similar repositories for hyper-nn

Users that are interested in hyper-nn are comparing it to the libraries listed below

• matthias-wright / flaxmodels
  Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
  ☆254Updated 3 months ago
• lucidrains / ponder-transformer
  Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper
  ☆81Updated 3 years ago
• yiyixuxu / denoising-diffusion-flax
  Implementing the Denoising Diffusion Probabilistic Model in Flax
  ☆150Updated 2 years ago
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆140Updated last year
• ag1988 / dss
  Sequence Modeling with Structured State Spaces
  ☆65Updated 2 years ago
• AlexiaJM / score_sde_fast_sampling
  Repository for the "Gotta Go Fast When Generating Data with Score-Based Models" paper
  ☆105Updated 3 years ago
• DSL-Lab / GRBM
  Gaussian-Bernoulli Restricted Boltzmann Machines
  ☆104Updated 2 years ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆112Updated 3 years ago
• kingoflolz / CLIP_JAX
  Contrastive Language-Image Pretraining
  ☆143Updated 2 years ago
• rasmusbergpalm / vnca
  ☆53Updated 2 years ago
• gisilvs / AEF
  ☆33Updated 2 years ago
• patil-suraj / stable-diffusion-jax
  ☆90Updated 2 years ago
• lucidrains / Mega-pytorch
  Implementation of Mega, the Single-head Attention with Multi-headed EMA architecture that currently holds SOTA on Long Range Arena
  ☆204Updated last year
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆179Updated last month
• CW-Huang / sdeflow-light
  A minimalist implementation of score-based diffusion model
  ☆127Updated 3 years ago
• google-deepmind / neural_testbed
  ☆192Updated 2 weeks ago
• tristandeleu / jax-meta-learning
  A collection of meta-learning algorithms in Jax
  ☆23Updated 2 years ago
• kvfrans / splus
  ☆110Updated last month
• radarFudan / mamba-minimal-jax
  ☆31Updated 7 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆84Updated last year
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆28Updated 11 months ago
• toshas / torch-discounted-cumsum
  Fast Discounted Cumulative Sums in PyTorch
  ☆96Updated 3 years ago
• AshishKumar4 / FlaxDiff
  A simple, easy-to-understand library for diffusion models using Flax and Jax. Includes detailed notebooks on DDPM, DDIM, and EDM with sim…
  ☆29Updated 2 months ago
• michaelsdr / sinkformers
  Transformers with doubly stochastic attention
  ☆46Updated 2 years ago
• thjashin / multires-conv
  Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)
  ☆124Updated last year
• lucidrains / gated-state-spaces-pytorch
  Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch
  ☆101Updated 2 years ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆90Updated last month
• young-geng / tpu_pod_commander
  TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.
  ☆20Updated last year
• Rayhane-mamah / Efficient-VDVAE
  Official Pytorch and JAX implementation of "Efficient-VDVAE: Less is more"
  ☆196Updated 2 years ago
• lucidrains / GAF-microbatch-pytorch
  Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch
  ☆25Updated 5 months ago