Alternatives and similar repositories for Flax-ResNets

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

• sjunhongshen / DASH
  ☆23Updated 2 years ago
• srush / mamba-scans
  Blog post
  ☆17Updated last year
• shikaiqiu / compute-better-spent
  ☆53Updated 8 months ago
• wjxts / RegularizedBN
  ☆21Updated 2 years ago
• xu-ji / information-bottleneck
  Deep Learning & Information Bottleneck
  ☆60Updated last year
• ahennequ / pytorch-custom-mma
  ☆29Updated 2 years ago
• JeanKaddour / LAWA
  Latest Weight Averaging (NeurIPS HITY 2022)
  ☆30Updated 2 years ago
• fKunstner / noise-sgd-adam-sign
  ☆16Updated 2 years ago
• formll / resolving-scaling-law-discrepancies
  ☆20Updated 11 months ago
• rwightman / imagenet-12k
  ImageNet-12k subset of ImageNet-21k (fall11)
  ☆21Updated 2 years ago
• y0ast / pytorch-snippets
  Collection of snippets for PyTorch users
  ☆25Updated 3 years ago
• AndyShih12 / LongHorizonTemperatureScaling
  PyTorch implementation for "Long Horizon Temperature Scaling", ICML 2023
  ☆20Updated 2 years ago
• tml-epfl / why-weight-decay
  Why Do We Need Weight Decay in Modern Deep Learning? [NeurIPS 2024]
  ☆66Updated 9 months ago
• JonasGeiping / dataaugs
  ☆18Updated 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
• dydjw9 / Efficient_SAM
  ☆58Updated 2 years ago
• tml-epfl / understanding-sam
  Towards Understanding Sharpness-Aware Minimization [ICML 2022]
  ☆35Updated 3 years ago
• hushon / JAX-ResNet-CIFAR10
  Simple CIFAR10 ResNet example with JAX.
  ☆23Updated 4 years ago
• SamsungSAILMontreal / ghn3
  Code for "Can We Scale Transformers to Predict Parameters of Diverse ImageNet Models?" [ICML 2023]
  ☆35Updated 10 months ago
• amirzandieh / HyperAttention
  Triton Implementation of HyperAttention Algorithm
  ☆48Updated last year
• hlml / fortuitous_forgetting
  ☆19Updated 3 years ago
• JeanKaddour / NoTrainNoGain
  Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)
  ☆80Updated last year
• tml-epfl / sharpness-vs-generalization
  A modern look at the relationship between sharpness and generalization [ICML 2023]
  ☆43Updated last year
• ondrejbohdal / meta-calibration
  Official PyTorch implementation of "Meta-Calibration: Learning of Model Calibration Using Differentiable Expected Calibration Error"
  ☆36Updated last year
• facebookresearch / ModelRatatouille
  Recycling diverse models
  ☆44Updated 2 years ago
• alexrame / diwa
  DiWA: Diverse Weight Averaging for Out-of-Distribution Generalization
  ☆31Updated 2 years ago
• ganguli-lab / degrees-of-freedom
  ☆37Updated 3 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 …
  ☆65Updated 4 years ago
• lucidrains / esbn-transformer
  An attempt to merge ESBN with Transformers, to endow Transformers with the ability to emergently bind symbols
  ☆16Updated 3 years ago
• minhtannguyen / SRSGD
  Code base for SRSGD.
  ☆28Updated 5 years ago