Create animations for the optimization trajectory of neural nets
☆163Jan 30, 2024Updated 2 years ago
Alternatives and similar repositories for loss-landscape-anim
Users that are interested in loss-landscape-anim are comparing it to the libraries listed below
Sorting:
- Approximating neural network loss landscapes in low-dimensional parameter subspaces for PyTorch☆351Nov 30, 2023Updated 2 years ago
- Visualizing the the loss landscape of Fully-Connected Neural Networks☆46Jun 4, 2023Updated 2 years ago
- Code for visualizing the loss landscape of neural nets☆3,157Apr 5, 2022Updated 3 years ago
- CIFAR10 ResNets implemented in JAX+Flax☆12Apr 6, 2022Updated 3 years ago
- ☆19Jul 15, 2020Updated 5 years ago
- This repo is built to facilitate the training and analysis of autoregressive transformers on maze-solving tasks.☆35Oct 28, 2025Updated 4 months ago
- LLA is a PyTorch library that allows to visualize and analyze loss landscapes of neural networks.☆13Dec 9, 2025Updated 3 months ago
- Notes about the intuition behind calculus on manifolds and the generalized Stokes's theorem☆22Feb 7, 2022Updated 4 years ago
- Efficient Neural Network Loss Landscape Generation☆10Apr 26, 2019Updated 6 years ago
- Explores the ideas presented in Deep Ensembles: A Loss Landscape Perspective (https://arxiv.org/abs/1912.02757) by Stanislav Fort, Huiyi …☆66Jul 29, 2020Updated 5 years ago
- An implementation of the penalty-based bilevel gradient descent (PBGD) algorithm and the iterative differentiation (ITD/RHG) methods.☆19Feb 13, 2023Updated 3 years ago
- Yet another mini autodiff system for educational purposes☆32Nov 11, 2025Updated 4 months ago
- Hypercorn is an ASGI and WSGI Server based on Hyper libraries and inspired by Gunicorn.☆14Jan 12, 2026Updated 2 months ago
- CoSCL: Cooperation of Small Continual Learners is Stronger than a Big One (ECCV 2022)☆19Sep 26, 2023Updated 2 years ago
- An unofficial implementation of the Fourier Neural Operator in Flax☆19May 25, 2024Updated last year
- the paper "Geometry-aware Instance-reweighted Adversarial Training" ICLR 2021 oral☆59Apr 13, 2021Updated 4 years ago
- Minimal implementation of adaptive gradient clipping (https://arxiv.org/abs/2102.06171) in TensorFlow 2.☆86Jun 16, 2021Updated 4 years ago
- ☆35Dec 5, 2022Updated 3 years ago
- Weight-Averaged Sharpness-Aware Minimization (NeurIPS 2022)☆28Jan 13, 2023Updated 3 years ago
- [ICML 2022]Source code for "A Closer Look at Smoothness in Domain Adversarial Training",☆72Apr 11, 2024Updated last year
- Customizable charts made with TikZ and LaTeX3☆14Feb 11, 2023Updated 3 years ago
- ☆23Nov 1, 2022Updated 3 years ago
- Source code for paper "Trajectory of Alternating Direction Method of Multipliers and Adaptive Acceleration" of NeurIPS 2019☆10Jan 25, 2024Updated 2 years ago
- Code to reproduce key results accompanying "SAEs (usually) Transfer Between Base and Chat Models"☆13Jul 18, 2024Updated last year
- A comprehensive framework for benchmarking single and multi-agent systems across a wide range of tasks—evaluating performance, accuracy, …☆36Nov 11, 2025Updated 4 months ago
- [CVPR 2022] "The Principle of Diversity: Training Stronger Vision Transformers Calls for Reducing All Levels of Redundancy" by Tianlong C…☆25Mar 9, 2022Updated 4 years ago
- SAM: Sharpness-Aware Minimization (PyTorch)☆1,965Feb 21, 2024Updated 2 years ago
- Codebase for " Reducing Representation Drift in Online Continual Learning"☆14Jun 8, 2021Updated 4 years ago
- ☆74Dec 7, 2024Updated last year
- Awesome papers in machine learning theory☆10Feb 12, 2022Updated 4 years ago
- A small game demonstrating a grid distortion effect☆15Oct 5, 2021Updated 4 years ago
- Neural Tangent Kernel Papers☆123Jan 12, 2025Updated last year
- ☆22May 27, 2018Updated 7 years ago
- ☆12Aug 21, 2020Updated 5 years ago
- ICML2019 Accepted Paper. Overcoming Multi-Model Forgetting☆14Jun 5, 2019Updated 6 years ago
- SC-Adagrad, SC-RMSProp and RMSProp algorithms for training deep networks proposed in☆14Oct 5, 2018Updated 7 years ago
- The official project website of "NORM: Knowledge Distillation via N-to-One Representation Matching" (The paper of NORM is published in IC…☆20Sep 18, 2023Updated 2 years ago
- Pruning methods for pytorch with an optimizer-like interface☆15Apr 14, 2020Updated 5 years ago
- ☆18Nov 8, 2023Updated 2 years ago