Alternatives and similar repositories for Flax-ResNets:

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

• sjunhongshen / DASH
  ☆21Updated 2 years ago
• JeanKaddour / LAWA
  Latest Weight Averaging (NeurIPS HITY 2022)
  ☆28Updated last year
• shikaiqiu / compute-better-spent
  ☆52Updated 4 months ago
• mansheej / lth_diet
  ☆10Updated 2 years ago
• ahennequ / pytorch-custom-mma
  ☆29Updated 2 years ago
• y0ast / pytorch-snippets
  Collection of snippets for PyTorch users
  ☆25Updated 2 years ago
• facebookresearch / projUNN
  Fast training of unitary deep network layers from low-rank updates
  ☆28Updated 2 years ago
• edwardjhu / TP4
  Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)
  ☆59Updated 3 years ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆82Updated last year
• xu-ji / information-bottleneck
  Deep Learning & Information Bottleneck
  ☆56Updated last year
• JonasGeiping / dataaugs
  ☆17Updated 2 years ago
• JeanKaddour / NoTrainNoGain
  Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)
  ☆79Updated last year
• tml-epfl / understanding-sam
  Towards Understanding Sharpness-Aware Minimization [ICML 2022]
  ☆35Updated 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
• formll / resolving-scaling-law-discrepancies
  ☆17Updated 7 months ago
• fKunstner / noise-sgd-adam-sign
  ☆15Updated last year
• gregorbachmann / scaling_mlps
  ☆51Updated 8 months ago
• srush / mamba-scans
  Blog post
  ☆16Updated last year
• AndyShih12 / LongHorizonTemperatureScaling
  PyTorch implementation for "Long Horizon Temperature Scaling", ICML 2023
  ☆20Updated last year
• js-d / sim_metric
  ☆35Updated last year
• aniquetahir / JORA
  JORA: JAX Tensor-Parallel LoRA Library (ACL 2024)
  ☆32Updated 9 months ago
• radarFudan / mamba-minimal-jax
  ☆30Updated 2 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 last year
• 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
• minhtannguyen / SRSGD
  Code base for SRSGD.
  ☆28Updated 4 years ago
• locuslab / edge-of-stability
  ☆63Updated 2 months ago
• davisyoshida / abnormal-floats
  Code for the note "NF4 Isn't Information Theoretically Optimal (and that's Good)
  ☆18Updated last year
• rwightman / imagenet-12k
  ImageNet-12k subset of ImageNet-21k (fall11)
  ☆21Updated last year
• SamsungSAILMontreal / ghn3
  Code for "Can We Scale Transformers to Predict Parameters of Diverse ImageNet Models?" [ICML 2023]
  ☆31Updated 5 months ago