A library to inspect and extract intermediate layers of PyTorch models.
☆473May 12, 2022Updated 4 years ago
Alternatives and similar repositories for surgeon-pytorch
Users that are interested in surgeon-pytorch are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- AI on the way. An auto deep learning pipe dream. An RDBMS approach to deep learning. Declarative, explainable, scalable, optimizable, eas…☆1,825Updated this week
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,484May 2, 2025Updated last year
- Cyclemoid implementation for PyTorch☆90Apr 2, 2022Updated 4 years ago
- FFCV: Fast Forward Computer Vision (and other ML workloads!)☆2,993Jun 16, 2024Updated 2 years ago
- Named tensors with first-class dimensions for PyTorch☆334Jun 14, 2023Updated 3 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.☆1,258Updated this week
- A collection of useful audio datasets and transforms for PyTorch.☆144Feb 11, 2023Updated 3 years ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,561Jul 5, 2026Updated 3 weeks ago
- 🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch☆2,195Jul 9, 2026Updated 2 weeks ago
- TheGlueNote is representation model for note-wise music alignment.☆14Jul 19, 2024Updated 2 years ago
- Audio generation using diffusion models, in PyTorch.☆2,095Jun 12, 2023Updated 3 years ago
- Tensors, for human consumption☆1,391Apr 9, 2026Updated 3 months 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
- Implementation of a Transformer using ReLA (Rectified Linear Attention) from https://arxiv.org/abs/2104.07012☆49Apr 6, 2022Updated 4 years ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- The WeightWatcher tool for predicting the accuracy of Deep Neural Networks☆1,769May 11, 2026Updated 2 months ago
- Task-driven models of the dorsal visual stream☆20Jun 15, 2026Updated last month
- Implementation of LogAvgExp for Pytorch☆37Apr 10, 2025Updated last year
- Collection of audio-focused loss functions in PyTorch☆874Jul 30, 2024Updated last year
- The correct way to resize images or tensors. For Numpy or Pytorch (differentiable).☆566Jul 13, 2023Updated 3 years ago
- A concise but complete full-attention transformer with a set of promising experimental features from various papers☆5,928Updated this week
- 🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…☆9,800Updated this 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
- functorch is JAX-like composable function transforms for PyTorch.☆1,434Aug 21, 2025Updated 11 months ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Supercharge Your Model Training☆5,490Apr 29, 2026Updated 2 months ago
- Trainer for audio-diffusion-pytorch☆129Jan 13, 2023Updated 3 years ago
- A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.☆3,149Feb 9, 2026Updated 5 months ago
- Library for Textless Spoken Language Processing☆559Aug 29, 2023Updated 2 years ago
- Aim 💫 — An easy-to-use & supercharged open-source experiment tracker.☆6,211Updated this week
- Model interpretability and understanding for PyTorch☆5,676Updated this week
- Image data augmentation scheduler for albumentations transforms☆19May 1, 2021Updated 5 years ago
- A data augmentations library for audio, image, text, and video.☆5,086Jul 16, 2026Updated last week
- Official repository with code and data accompanying the NAACL 2021 paper "Hurdles to Progress in Long-form Question Answering" (https://a…☆46Jul 30, 2022Updated 3 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Transformer-based visually grounded speech models☆19Sep 22, 2022Updated 3 years ago
- Python scripts to create noisy and reverberant 2-speaker mixture audio with Libri-Light and WHAM☆17Nov 7, 2024Updated last year
- Fast audio data augmentation in PyTorch. Inspired by audiomentations. Useful for deep learning.☆1,162Nov 24, 2025Updated 8 months ago
- TensorDict is a pytorch dedicated tensor container.☆1,034Updated this week
- A python library for self-supervised learning on images.☆3,788Updated this week
- 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 4 years ago
- ☆64Dec 15, 2023Updated 2 years ago