f-dangel / cockpitView external linksLinks
Cockpit: A Practical Debugging Tool for Training Deep Neural Networks
☆488Jul 1, 2022Updated 3 years ago
Alternatives and similar repositories for cockpit
Users that are interested in cockpit are comparing it to the libraries listed below
Sorting:
- Experiments for the NeurIPS 2021 paper "Cockpit: A Practical Debugging Tool for the Training of Deep Neural Networks"☆13Oct 25, 2021Updated 4 years ago
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Jul 19, 2023Updated 2 years ago
- BackPACK - a backpropagation package built on top of PyTorch which efficiently computes quantities other than the gradient.☆604Nov 28, 2025Updated 2 months ago
- The WeightWatcher tool for predicting the accuracy of Deep Neural Networks☆1,721Dec 12, 2025Updated 2 months ago
- Lightweight ML Experiment Logging 📖☆80Aug 26, 2024Updated last year
- Lightweight Cluster/Cloud VM Job Management 🚀☆42Aug 27, 2024Updated last year
- Code for the anonymous submission "Cockpit: A Practical Debugging Tool for Training Deep Neural Networks"☆31Nov 24, 2020Updated 5 years ago
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,471May 2, 2025Updated 9 months ago
- This repository contains the results for the paper: "Descending through a Crowded Valley - Benchmarking Deep Learning Optimizers"☆184Jul 17, 2021Updated 4 years ago
- ☆60Mar 8, 2022Updated 3 years ago
- Lightweight Hyperparameter Optimization 🚂☆149Aug 27, 2024Updated last year
- Lightweight Experiment & Resource Monitoring 📺☆187Mar 8, 2023Updated 2 years ago
- Implicit MLE: Backpropagating Through Discrete Exponential Family Distributions☆259Oct 29, 2023Updated 2 years ago
- Code for "Self-Attention Between Datapoints: Going Beyond Individual Input-Output Pairs in Deep Learning"☆415Mar 21, 2024Updated last year
- An open source implementation of CLIP.☆33Nov 7, 2022Updated 3 years ago
- Toolbox of models, callbacks, and datasets for AI/ML researchers.☆1,755Jan 20, 2026Updated 3 weeks ago
- A library for distributed ML training with PyTorch☆366Dec 12, 2022Updated 3 years ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,161Mar 22, 2024Updated last year
- A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.☆1,248Updated this week
- FFCV: Fast Forward Computer Vision (and other ML workloads!)☆2,989Jun 16, 2024Updated last year
- Code for paper "SurVAE Flows: Surjections to Bridge the Gap between VAEs and Flows"☆290Feb 1, 2021Updated 5 years ago
- PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆331Jun 13, 2022Updated 3 years ago
- Code for: "Neural Rough Differential Equations for Long Time Series", (ICML 2021)☆122May 11, 2021Updated 4 years ago
- VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.☆3,293Mar 3, 2024Updated last year
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,395Jan 26, 2026Updated 2 weeks ago
- ☆133Mar 23, 2021Updated 4 years ago
- Fast, differentiable sorting and ranking in PyTorch☆850Jun 9, 2025Updated 8 months ago
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,629Mar 25, 2022Updated 3 years ago
- Version control for machine learning☆1,671Feb 25, 2025Updated 11 months ago
- Your PyTorch AI Factory - Flash enables you to easily configure and run complex AI recipes for over 15 tasks across 7 data domains☆1,736Oct 8, 2023Updated 2 years ago
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆310Jan 12, 2026Updated last month
- PyTorch extensions for high performance and large scale training.☆3,397Apr 26, 2025Updated 9 months ago
- A collection of code snippets for my PyTorch Lightning projects☆107Jan 18, 2021Updated 5 years ago
- Fast and Easy Infinite Neural Networks in Python☆2,374Mar 1, 2024Updated last year
- Tensors, for human consumption☆1,353Jan 22, 2026Updated 3 weeks ago
- The spiritual successor to knockknock for PyTorch Lightning, get notified when your training ends☆77Dec 4, 2024Updated last year
- Implementation of the models and datasets used in "An Information-theoretic Approach to Distribution Shifts"☆25Nov 2, 2021Updated 4 years ago
- Live Python Notebooks with any Editor☆279Dec 7, 2022Updated 3 years ago
- An unopinionated replacement for PyTorch's Dataset and ImageFolder, that handles Tar archives☆78Aug 10, 2022Updated 3 years ago