A library to inspect and extract intermediate layers of PyTorch models.
☆474May 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,824Updated this week
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,482May 2, 2025Updated last year
- Cyclemoid implementation for PyTorch☆90Apr 2, 2022Updated 4 years ago
- FFCV: Fast Forward Computer Vision (and other ML workloads!)☆2,989Jun 16, 2024Updated last year
- Named tensors with first-class dimensions for PyTorch☆331Jun 14, 2023Updated 3 years ago
- Open source password manager - Proton Pass • AdSecurely store, share, and autofill your credentials with Proton Pass, the end-to-end encrypted password manager trusted by millions.
- A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.☆1,255Updated 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,507May 31, 2026Updated 2 weeks ago
- 🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch☆2,189Nov 27, 2024Updated last year
- TheGlueNote is representation model for note-wise music alignment.☆14Jul 19, 2024Updated last year
- Audio generation using diffusion models, in PyTorch.☆2,099Jun 12, 2023Updated 3 years ago
- Tensors, for human consumption☆1,387Apr 9, 2026Updated 2 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
- 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.
- The WeightWatcher tool for predicting the accuracy of Deep Neural Networks☆1,756May 11, 2026Updated last month
- Implementation of LogAvgExp for Pytorch☆37Apr 10, 2025Updated last year
- Collection of audio-focused loss functions in PyTorch☆870Jul 30, 2024Updated last year
- The correct way to resize images or tensors. For Numpy or Pytorch (differentiable).☆566Jul 13, 2023Updated 2 years ago
- A concise but complete full-attention transformer with a set of promising experimental features from various papers☆5,893Jun 8, 2026Updated last week
- 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 simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…☆9,720Updated 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
- A spoken version of the textual story cloze benchmark☆22Aug 6, 2023Updated 2 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- functorch is JAX-like composable function transforms for PyTorch.☆1,436Aug 21, 2025Updated 9 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,114Feb 9, 2026Updated 4 months ago
- Library for Textless Spoken Language Processing☆558Aug 29, 2023Updated 2 years ago
- Model interpretability and understanding for PyTorch☆5,647Jun 1, 2026Updated 2 weeks ago
- Aim 💫 — An easy-to-use & supercharged open-source experiment tracker.☆6,155Updated this week
- Image data augmentation scheduler for albumentations transforms☆19May 1, 2021Updated 5 years 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 3 years ago
- A data augmentations library for audio, image, text, and video.☆5,084Jun 1, 2026Updated 2 weeks ago
- End-to-end encrypted cloud storage - Proton Drive • AdSpecial offer: 40% Off Yearly / 80% Off First Month. Protect your most important files, photos, and documents from prying eyes.
- 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
- TensorDict is a pytorch dedicated tensor container.