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,995Jun 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.
- Tensors, for human consumption☆1,399Apr 9, 2026Updated 6 months ago
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,627Aug 26, 2026Updated 2 months ago
- A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.☆1,264Sep 16, 2026Updated last month
- Audio generation using diffusion models, in PyTorch.☆2,098Jun 12, 2023Updated 3 years ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,434Aug 21, 2025Updated last year
- Implementation of LogAvgExp for Pytorch☆37Apr 10, 2025Updated last year
- Supercharge Your Model Training☆5,508Apr 29, 2026Updated 6 months ago
- 🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch☆2,206Jul 9, 2026Updated 3 months ago
- The correct way to resize images or tensors. For Numpy or Pytorch (differentiable).☆567Jul 13, 2023Updated 3 years ago
- 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.
- A concise but complete implementation of CLIP with various experimental improvements from recent papers☆725Oct 16, 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
- A collection of useful audio datasets and transforms for PyTorch.☆144Feb 11, 2023Updated 3 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 4 years ago
- The WeightWatcher tool for predicting the accuracy of Deep Neural Networks☆1,778Updated this week
- PyTorch – SMART: Robust and Efficient Fine-Tuning for Pre-trained Natural Language Models.☆62Jun 28, 2022Updated 4 years ago
- A concise but complete full-attention transformer with a set of promising experimental features from various papers☆5,961Oct 23, 2026Updated last week
- TensorDict is a pytorch dedicated tensor container.☆1,043Updated this week
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆252Sep 1, 2022Updated 4 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- 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
- Trainer for audio-diffusion-pytorch☆128Jan 13, 2023Updated 3 years ago
- VISSL is FAIR's library of extensible, modular and scalable components for SOTA Self-Supervised Learning with images.☆3,289Mar 3, 2024Updated 2 years ago
- A data augmentations library for audio, image, text, and video.☆5,098Oct 20, 2026Updated last week
- Fast audio data augmentation in PyTorch. Inspired by audiomentations. Useful for deep learning.☆1,174Nov 24, 2025Updated 11 months ago
- 🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…☆9,930Updated this week
- A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.☆3,212Feb 9, 2026Updated 8 months ago
- Model interpretability and understanding for PyTorch☆5,719Updated this week
- Version control for machine learning☆1,680Feb 25, 2025Updated last year
- 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.
- Library for 8-bit optimizers and quantization routines.☆777Aug 18, 2022Updated 4 years ago
- Cockpit: A Practical Debugging Tool for Training Deep Neural Networks☆491Jul 1, 2022Updated 4 years ago
- Virtual Adversarial Training (VAT) techniques in PyTorch☆17Jul 19, 2022Updated 4 years ago
- ☆217Oct 10, 2022Updated 4 years ago
- PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code☆697Mar 25, 2023Updated 3 years ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,172Mar 22, 2024Updated 2 years ago
- Collection of audio-focused loss functions in PyTorch☆890Jul 30, 2024Updated 2 years ago