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 10 months ago
• jiaweizzhao / ZerO-initialization
  ☆75Updated 2 years ago
• js-d / sim_metric
  ☆37Updated 2 years ago
• JeanKaddour / NoTrainNoGain
  Revisiting Efficient Training Algorithms For Transformer-based Language Models (NeurIPS 2023)
  ☆80Updated 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 …
  ☆66Updated 5 years ago
• shikaiqiu / compute-better-spent
  ☆58Updated last year
• LIONS-EPFL / scion
  ☆45Updated last week
• irhum / esmjax
  ESM2 protein language models in JAX/Flax
  ☆17Updated 3 years ago
• hlml / fortuitous_forgetting
  ☆19Updated 3 years ago
• srush / mamba-scans
  Blog post
  ☆17Updated last year
• Sea-Snell / grokking
  unofficial re-implementation of "Grokking: Generalization Beyond Overfitting on Small Algorithmic Datasets"
  ☆79Updated 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
  ☆88Updated last year
• JeanKaddour / LAWA
  Latest Weight Averaging (NeurIPS HITY 2022)
  ☆31Updated 2 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
• epfml / schedules-and-scaling
  Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"
  ☆84Updated last year
• 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
• srush / do-we-need-attention
  ☆166Updated 2 years ago
• wjxts / RegularizedBN
  ☆21Updated 2 years ago
• epfml / llm-baselines
  nanoGPT-like codebase for LLM training
  ☆109Updated 5 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…
  ☆40Updated 2 years ago
• ethancaballero / broken_neural_scaling_laws
  Code Release for "Broken Neural Scaling Laws" (BNSL) paper
  ☆59Updated 2 years ago
• lucidrains / infini-transformer-pytorch
  Implementation of Infini-Transformer in Pytorch
  ☆113Updated 9 months ago
• google-deepmind / eigengame
  Open source code for EigenGame.
  ☆33Updated 2 years ago
• apple / learning-subspaces
  ☆133Updated 4 years 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 4 years ago
• swyoon / pytorch-energy-based-model
  Simple illustrative examples for energy-based models in PyTorch
  ☆66Updated 5 years ago
• ml-jku / EVA
  One Initialization to Rule them All: Fine-tuning via Explained Variance Adaptation
  ☆44Updated last week
• stanislavfort / dissect-git-re-basin
  Replicating and dissecting the git-re-basin project in one-click-replication Colabs
  ☆35Updated 3 years ago
• facebookresearch / ModelRatatouille
  Recycling diverse models
  ☆46Updated 2 years ago