Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory
☆434Aug 29, 2024Updated 2 years ago
Alternatives and similar repositories for toma
Users that are interested in toma are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,485May 2, 2025Updated last year
- High performance model preprocessing library on PyTorch☆639Mar 29, 2024Updated 2 years ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,434Aug 21, 2025Updated last year
- 🚀 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
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆9,626Aug 26, 2026Updated 2 months ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.☆1,265Sep 16, 2026Updated last month
- PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆331Jun 13, 2022Updated 4 years ago
- FFCV: Fast Forward Computer Vision (and other ML workloads!)☆2,995Jun 16, 2024Updated 2 years ago
- ☆154May 25, 2020Updated 6 years ago
- Profiling and inspecting memory in pytorch☆1,079Aug 12, 2026Updated 2 months ago
- A set of utilities to turn Dataclasses into useful configuration managers.☆11Mar 27, 2024Updated 2 years ago
- PyTorch extensions for fast R&D prototyping and Kaggle farming☆1,575Oct 9, 2025Updated last year
- Toolbox of models, callbacks, and datasets for AI/ML researchers.☆1,753Jan 20, 2026Updated 9 months ago
- 🚪✊Knock Knock: Get notified when your training ends with only two additional lines of code☆2,824Jun 23, 2023Updated 3 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.
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,624Mar 25, 2022Updated 4 years ago
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,078Apr 17, 2024Updated 2 years ago
- Implementation of FixMatch in PyTorch and experimentations☆12Aug 9, 2020Updated 6 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
- TensorDict is a pytorch dedicated tensor container.☆1,043Updated this week
- Version control for machine learning☆1,680Feb 25, 2025Updated last year
- 🐍 Geometric Computer Vision Library for Spatial AI☆11,455Updated this week
- Hackable and optimized Transformers building blocks, supporting a composable construction.☆10,572Updated this week
- Shape and dimension inference (Keras-like) for PyTorch layers and neural networks☆572Jun 13, 2022Updated 4 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- A Data Streaming Library for Efficient Neural Network Training☆1,571Jun 25, 2026Updated 4 months ago
- Provides everything needed for high performance data loading and augmentation in pytorch.☆320Jan 1, 2024Updated 2 years ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,172Mar 22, 2024Updated 2 years ago
- PyTorch layer-by-layer model profiler☆604May 23, 2021Updated 5 years ago
- ☆29Nov 30, 2021Updated 4 years ago
- AutoML for image augmentation. AutoAlbument uses the Faster AutoAugment algorithm to find optimal augmentation policies. Documentation - …☆206Sep 4, 2021Updated 5 years ago
- Experimental framework for running pytorch experiments☆14Mar 6, 2023Updated 3 years ago
- PyTorch elastic training☆726Jun 15, 2022Updated 4 years ago
- Model interpretability and understanding for PyTorch☆5,717Oct 18, 2026Updated last week
- 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.
- ☆16Apr 21, 2022Updated 4 years ago
- torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters i…☆179Dec 16, 2025Updated 10 months ago
- Python 3.8+ toolbox for submitting jobs to Slurm☆1,642Jan 14, 2026Updated 9 months ago
- Course webpage for COMP 790, (Deep) Learning from Limited Labeled Data☆304Aug 13, 2020Updated 6 years ago
- 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
- PyTorch extensions for high performance and large scale training.☆3,404Apr 26, 2025Updated last year
- Fine-Tuning Pre-trained Transformers into Decaying Fast Weights☆20Oct 9, 2022Updated 4 years ago