archinetai / surgeon-pytorchView external linksLinks
A library to inspect and extract intermediate layers of PyTorch models.
☆476May 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,832Jan 18, 2026Updated 3 weeks ago
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,473May 2, 2025Updated 9 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,989Jun 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,248Updated this week
- A minimalistic boiler plate code for training pytorch models☆13Jun 10, 2024Updated last year
- 🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch☆2,184Nov 27, 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
- Implementation of a Transformer using ReLA (Rectified Linear Attention) from https://arxiv.org/abs/2104.07012☆49Apr 6, 2022Updated 3 years ago
- Tensors, for human consumption☆1,353Jan 22, 2026Updated 3 weeks ago
- Implementation of Mega, the Single-head Attention with Multi-headed EMA architecture that currently holds SOTA on Long Range Arena☆207Aug 26, 2023Updated 2 years ago
- Audio generation using diffusion models, in PyTorch.☆2,096Jun 12, 2023Updated 2 years ago
- A concise but complete full-attention transformer with a set of promising experimental features from various papers☆5,800Updated this week
- The WeightWatcher tool for predicting the accuracy of Deep Neural Networks☆1,718Dec 12, 2025Updated 2 months ago
- 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
- A concise but complete implementation of CLIP with various experimental improvements from recent papers☆722Oct 16, 2023Updated 2 years ago
- Collection of audio-focused loss functions in PyTorch☆846Jul 30, 2024Updated last year
- For optimization algorithm research and development.☆558Jan 12, 2026Updated last month
- 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,067Updated this week
- 🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…☆9,491Feb 6, 2026Updated last week
- The correct way to resize images or tensors. For Numpy or Pytorch (differentiable).☆564Jul 13, 2023Updated 2 years ago
- Aim 💫 — An easy-to-use & supercharged open-source experiment tracker.☆5,992Updated this week
- functorch is JAX-like composable function transforms for PyTorch.☆1,436Aug 21, 2025Updated 5 months ago
- Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch☆59Jan 13, 2021Updated 5 years ago
- Model interpretability and understanding for PyTorch☆5,556Feb 3, 2026Updated last week
- Version control for machine learning☆1,671Feb 25, 2025Updated 11 months ago
- A python library for self-supervised learning on images.☆3,679Dec 19, 2025Updated last month
- A spoken version of the textual story cloze benchmark☆20Aug 6, 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 lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…☆66Jul 31, 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,629Mar 25, 2022Updated 3 years ago
- Fast audio data augmentation in PyTorch. Inspired by audiomentations. Useful for deep learning.☆1,132Nov 24, 2025Updated 2 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.☆2,990Updated this week
- 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