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…
  ☆39Updated 8 months ago
• LIONS-EPFL / scion
  ☆37Updated 3 weeks ago
• y0ast / pytorch-snippets
  Collection of snippets for PyTorch users
  ☆25Updated 3 years ago
• edwardjhu / TP4
  Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)
  ☆62Updated 4 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…
  ☆83Updated 2 months 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 5 years ago
• JeanKaddour / NoTrainNoGain
  Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)
  ☆81Updated 2 years ago
• apple / learning-subspaces
  ☆133Updated 4 years ago
• locuslab / edge-of-stability
  ☆70Updated 8 months ago
• hlml / fortuitous_forgetting
  ☆19Updated 3 years ago
• tml-epfl / understanding-sam
  Towards Understanding Sharpness-Aware Minimization [ICML 2022]
  ☆35Updated 3 years ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆87Updated last year
• lucidrains / infini-transformer-pytorch
  Implementation of Infini-Transformer in Pytorch
  ☆111Updated 7 months ago
• srush / mamba-scans
  Blog post
  ☆17Updated last year
• facebookresearch / ModelRatatouille
  Recycling diverse models
  ☆45Updated 2 years ago
• google-research / jax-influence
  ☆60Updated 3 years ago
• srush / do-we-need-attention
  ☆166Updated 2 years ago
• wjxts / RegularizedBN
  ☆21Updated 2 years ago
• AnanyaKumar / transfer_learning
  Framework code with wandb, checkpointing, logging, configs, experimental protocols. Useful for fine-tuning models or training from scratc…
  ☆151Updated 2 years ago
• epfml / llm-baselines
  nanoGPT-like codebase for LLM training
  ☆103Updated 3 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 …
  ☆40Updated 4 years ago
• gregorbachmann / scaling_mlps
  ☆51Updated last year
• dydjw9 / Efficient_SAM
  ☆58Updated 2 years ago
• sjunhongshen / DASH
  ☆23Updated 2 years ago
• tml-epfl / why-weight-decay
  Why Do We Need Weight Decay in Modern Deep Learning? [NeurIPS 2024]
  ☆67Updated 11 months ago
• jiaweizzhao / ZerO-initialization
  ☆75Updated 2 years ago
• mlfoundations / scaling
  Language models scale reliably with over-training and on downstream tasks
  ☆98Updated last year
• MadryLab / modeldiff
  ModelDiff: A Framework for Comparing Learning Algorithms
  ☆59Updated 2 years ago
• hsouri / BayesianTransferLearning
  ☆108Updated 2 years ago
• oripress / EntropyEnigma
  Official code for the ICML 2024 paper "The Entropy Enigma: Success and Failure of Entropy Minimization"
  ☆53Updated last year