A library to inspect and extract intermediate layers of PyTorch models.
☆473May 12, 2022Updated 3 years ago
Alternatives and similar repositories for surgeon-pytorch
Users that are interested in surgeon-pytorch are comparing it to the libraries listed below
Sorting:
- Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.☆1,829Jan 18, 2026Updated last month
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,471May 2, 2025Updated 10 months ago
- Cyclemoid implementation for PyTorch☆90Apr 2, 2022Updated 3 years ago
- Named tensors with first-class dimensions for PyTorch☆332Jun 14, 2023Updated 2 years ago
- FFCV: Fast Forward Computer Vision (and other ML workloads!)☆2,985Jun 16, 2024Updated last year
- A collection of useful audio datasets and transforms for PyTorch.☆144Feb 11, 2023Updated 3 years ago
- A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.☆1,249Updated this week
- A minimalistic boiler plate code for training pytorch models☆13Jun 10, 2024Updated last year
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,415Feb 20, 2026Updated 2 weeks ago
- Tensors, for human consumption☆1,360Jan 22, 2026Updated last month
- Audio generation using diffusion models, in PyTorch.☆2,094Jun 12, 2023Updated 2 years ago
- The WeightWatcher tool for predicting the accuracy of Deep Neural Networks☆1,726Feb 26, 2026Updated last week
- TheGlueNote is representation model for note-wise music alignment.☆12Jul 19, 2024Updated last year
- This is a intuitive explanation of Representation Learning with Contrastive Predictive Coding using code provided by jefflai108 that use…☆10Jan 25, 2021Updated 5 years ago
- Collection of audio-focused loss functions in PyTorch☆854Jul 30, 2024Updated last year
- For optimization algorithm research and development.☆557Updated this week
- Cockpit: A Practical Debugging Tool for Training Deep Neural Networks☆488Jul 1, 2022Updated 3 years ago
- A data augmentations library for audio, image, text, and video.☆5,071Feb 13, 2026Updated 3 weeks ago
- 🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…☆9,528Updated this week
- The correct way to resize images or tensors. For Numpy or Pytorch (differentiable).☆563Jul 13, 2023Updated 2 years ago
- Aim 💫 — An easy-to-use & supercharged open-source experiment tracker.☆6,017Updated this week
- functorch is JAX-like composable function transforms for PyTorch.☆1,436Aug 21, 2025Updated 6 months ago
- Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch☆58Jan 13, 2021Updated 5 years ago
- Model interpretability and understanding for PyTorch☆5,564Feb 26, 2026Updated last week
- Version control for machine learning☆1,670Feb 25, 2025Updated last year
- A python library for self-supervised learning on images.☆3,690Feb 23, 2026Updated last week
- A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…☆66Jul 31, 2023Updated 2 years ago
- Python scripts to create noisy and reverberant 2-speaker mixture audio with Libri-Light and WHAM☆17Nov 7, 2024Updated last year
- A spoken version of the textual story cloze benchmark☆20Aug 6, 2023Updated 2 years ago
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,627Mar 25, 2022Updated 3 years ago
- Fast audio data augmentation in PyTorch. Inspired by audiomentations. Useful for deep learning.☆1,135Nov 24, 2025Updated 3 months ago
- Library for Textless Spoken Language Processing☆555Aug 29, 2023Updated 2 years ago
- A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.☆3,004Feb 9, 2026Updated 3 weeks 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,731Oct 8, 2023Updated 2 years ago
- Trainer for audio-diffusion-pytorch☆129Jan 13, 2023Updated 3 years ago
- Deepchecks: Tests for Continuous Validation of ML Models & Data. Deepchecks is a holistic open-source solution for all of your AI & ML va…☆3,982Dec 28, 2025Updated 2 months ago
- Official repository with code and data accompanying the NAACL 2021 paper "Hurdles to Progress in Long-form Question Answering" (https://a…☆45Jul 30, 2022Updated 3 years ago
- RUDOLPH: One Hyper-Tasking Transformer can be creative as DALL-E and GPT-3 and smart as CLIP☆253Feb 6, 2023Updated 3 years ago
- PyTorch extensions for high performance and large scale training.☆3,400Apr 26, 2025Updated 10 months ago