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,823Updated this week
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,485May 2, 2025Updated last year
- Cyclemoid implementation for PyTorch☆90Apr 2, 2022Updated 4 years ago
- FFCV: Fast Forward Computer Vision (and other ML workloads!)☆2,992Jun 16, 2024Updated 2 years ago
- Named tensors with first-class dimensions for PyTorch☆334Jun 14, 2023Updated 3 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.☆1,267Updated 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,604Aug 26, 2026Updated 3 weeks ago
- 🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch☆2,204Jul 9, 2026Updated 2 months 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,395Apr 9, 2026Updated 5 months ago
- Implementation of Mega, the Single-head Attention with Multi-headed EMA architecture that currently holds SOTA on Long Range Arena☆206Aug 26, 2023Updated 3 years ago
- Implementation of a Transformer using ReLA (Rectified Linear Attention) from https://arxiv.org/abs/2104.07012☆49Apr 6, 2022Updated 4 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.
- The WeightWatcher tool for predicting the accuracy of Deep Neural Networks☆1,777May 11, 2026Updated 4 months ago
- Task-driven models of the dorsal visual stream☆20Jun 15, 2026Updated 3 months ago
- Implementation of LogAvgExp for Pytorch☆37Apr 10, 2025Updated last year
- Collection of audio-focused loss functions in PyTorch☆884Jul 30, 2024Updated 2 years ago
- 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,949Updated this week
- 🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…☆9,874Updated this week
- A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-…☆66Jul 31, 2023Updated 3 years ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,434Aug 21, 2025Updated last year
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Supercharge Your Model Training☆5,495Apr 29, 2026Updated 4 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,196Feb 9, 2026Updated 7 months ago
- Library for Textless Spoken Language Processing☆559Aug 29, 2023Updated 3 years ago
- Aim 💫 — An easy-to-use & supercharged open-source experiment tracker.☆6,256Updated this week
- Model interpretability and understanding for PyTorch☆5,701Sep 13, 2026Updated last 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,096Aug 24, 2026Updated 3 weeks ago
- Official repository with code and data accompanying the NAACL 2021 paper "Hurdles to Progress in Long-form Question Answering" (https://a…☆46Jul 30, 2022Updated 4 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- 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,168Nov 24, 2025Updated 9 months ago
- TensorDict is a pytorch dedicated tensor container.☆1,036Updated this week
- A python library for self-supervised learning on images.☆3,812Updated this week
- ☆64Dec 15, 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,625Mar 25, 2022Updated 4 years ago