Alternatives and similar repositories for Flax-ResNets

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

• facebookresearch / Pitfalls-of-Memorization
  Understanding the interplay between memorization and generalization in neural networks, featuring MAT, a learning algorithm to enhance ro…
  ☆40Updated 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 …
  ☆66Updated 5 years ago
• srush / mamba-scans
  Blog post
  ☆17Updated last year
• JeanKaddour / NoTrainNoGain
  Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)
  ☆81Updated 2 years ago
• tml-epfl / understanding-sam
  Towards Understanding Sharpness-Aware Minimization [ICML 2022]
  ☆38Updated 3 years ago
• edwardjhu / TP4
  Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)
  ☆63Updated 4 years ago
• fjzzq2002 / random_transformers
  Official code for "Algorithmic Capabilities of Random Transformers" (NeurIPS 2024)
  ☆16Updated last year
• sjunhongshen / DASH
  ☆23Updated 2 years ago
• js-d / sim_metric
  ☆37Updated 2 years ago
• gregorbachmann / scaling_mlps
  ☆52Updated last year
• LIONS-EPFL / scion
  ☆52Updated 3 weeks ago
• jiaweizzhao / ZerO-initialization
  ☆75Updated 3 years ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆92Updated last year
• epfml / llm-baselines
  nanoGPT-like codebase for LLM training
  ☆114Updated 2 months ago
• gibipara92 / learning-explanations-hard-to-vary
  Code to implement the AND-mask and geometric mean to do gradient based optimization, from the paper "Learning explanations that are hard …
  ☆41Updated 5 years ago
• shikaiqiu / compute-better-spent
  ☆62Updated last year
• facebookresearch / ModelRatatouille
  Recycling diverse models
  ☆46Updated 2 years ago
• JeanKaddour / LAWA
  Latest Weight Averaging (NeurIPS HITY 2022)
  ☆32Updated 2 years 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…
  ☆40Updated 2 years ago
• logancyang / loss-landscape-anim
  Create animations for the optimization trajectory of neural nets
  ☆161Updated last year
• google-research / jax-influence
  ☆63Updated 3 years ago
• ethancaballero / broken_neural_scaling_laws
  Code Release for "Broken Neural Scaling Laws" (BNSL) paper
  ☆59Updated 2 years ago
• epfml / schedules-and-scaling
  Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"
  ☆87Updated last year
• wjxts / RegularizedBN
  ☆21Updated 3 years ago
• MadryLab / modeldiff
  ModelDiff: A Framework for Comparing Learning Algorithms
  ☆58Updated 2 years ago
• google-deepmind / distribution_shift_framework
  This repository contains the code of the distribution shift framework presented in A Fine-Grained Analysis on Distribution Shift (Wiles e…
  ☆84Updated last month
• apple / learning-subspaces
  ☆133Updated 4 years ago
• tml-epfl / why-weight-decay
  Why Do We Need Weight Decay in Modern Deep Learning? [NeurIPS 2024]
  ☆70Updated last year
• y0ast / pytorch-snippets
  Collection of snippets for PyTorch users
  ☆25Updated 3 years ago
• aniquetahir / JORA
  JORA: JAX Tensor-Parallel LoRA Library (ACL 2024)
  ☆36Updated last year