Alternatives and similar repositories for Flax-ResNets:

Users that are interested in Flax-ResNets are comparing it to the libraries listed below

• fKunstner / noise-sgd-adam-sign
  ☆16Updated last year
• ahennequ / pytorch-custom-mma
  ☆29Updated 2 years ago
• sjunhongshen / DASH
  ☆22Updated 2 years ago
• shikaiqiu / compute-better-spent
  ☆52Updated 5 months ago
• srush / mamba-scans
  Blog post
  ☆17Updated last year
• SamsungSAILMontreal / ghn3
  Code for "Can We Scale Transformers to Predict Parameters of Diverse ImageNet Models?" [ICML 2023]
  ☆31Updated 7 months ago
• facebookresearch / projUNN
  Fast training of unitary deep network layers from low-rank updates
  ☆29Updated 2 years ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆83Updated last year
• y0ast / pytorch-snippets
  Collection of snippets for PyTorch users
  ☆25Updated 3 years ago
• formll / resolving-scaling-law-discrepancies
  ☆18Updated 8 months ago
• radarFudan / mamba-minimal-jax
  ☆30Updated 4 months ago
• JeanKaddour / LAWA
  Latest Weight Averaging (NeurIPS HITY 2022)
  ☆29Updated last year
• google-deepmind / eigengame
  Open source code for EigenGame.
  ☆30Updated last year
• young-geng / mlxu
  Machine Learning eXperiment Utilities
  ☆46Updated 9 months ago
• gregorbachmann / scaling_mlps
  ☆51Updated 9 months ago
• Ping-C / optimizer
  This repository includes code to reproduce the tables in "Loss Landscapes are All You Need: Neural Network Generalization Can Be Explaine…
  ☆35Updated 2 years ago
• amirzandieh / HyperAttention
  Triton Implementation of HyperAttention Algorithm
  ☆47Updated last year
• dasayan05 / iclr24_blog_code
  Code repo for ICLR 24 BlogPost titled "Building Diffusion Model's theory from ground up"
  ☆18Updated last year
• nick11roberts / XD
  ☆11Updated 2 years ago
• stanislavfort / dissect-git-re-basin
  Replicating and dissecting the git-re-basin project in one-click-replication Colabs
  ☆36Updated 2 years ago
• ayulockin / LossLandscape
  Explores the ideas presented in Deep Ensembles: A Loss Landscape Perspective (https://arxiv.org/abs/1912.02757) by Stanislav Fort, Huiyi …
  ☆63Updated 4 years ago
• aniquetahir / JORA
  JORA: JAX Tensor-Parallel LoRA Library (ACL 2024)
  ☆33Updated 11 months ago
• ganguli-lab / degrees-of-freedom
  ☆36Updated 3 years ago
• facebookresearch / ModelRatatouille
  Recycling diverse models
  ☆44Updated 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
  ☆23Updated 2 months ago
• alexrame / diwa
  DiWA: Diverse Weight Averaging for Out-of-Distribution Generalization
  ☆29Updated 2 years ago
• rwightman / imagenet-12k
  ImageNet-12k subset of ImageNet-21k (fall11)
  ☆21Updated last year
• Sanaelotfi / Bayesian_model_comparison
  Supporing code for the paper "Bayesian Model Selection, the Marginal Likelihood, and Generalization".
  ☆35Updated 2 years ago
• SamsungSAILMontreal / nino
  Code for "Accelerating Training with Neuron Interaction and Nowcasting Networks" [to appear at ICLR 2025]
  ☆18Updated 2 weeks ago
• mansheej / lth_diet
  ☆10Updated 2 years ago