pytorch / torchcodec
PyTorch video decoding
☆227Updated this week
Alternatives and similar repositories for torchcodec:
Users that are interested in torchcodec are comparing it to the libraries listed below
- Scalable and Performant Data Loading☆210Updated this week
- PyTorch per step fault tolerance (actively under development)☆226Updated this week
- Transform datasets at scale. Optimize datasets for fast AI model training.☆406Updated this week
- For optimization algorithm research and development.☆486Updated last week
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆511Updated this week
- ☆296Updated 7 months ago
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆344Updated 2 months ago
- This repository contains the experimental PyTorch native float8 training UX☆219Updated 5 months ago
- Best practices & guides on how to write distributed pytorch training code☆342Updated this week
- When it comes to optimizers, it's always better to be safe than sorry☆166Updated this week
- ☆121Updated this week
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆216Updated this week
- Helpful tools and examples for working with flex-attention☆603Updated this week
- DeMo: Decoupled Momentum Optimization☆171Updated last month
- FlashRNN - Fast RNN Kernels with I/O Awareness☆71Updated last month
- A library for unit scaling in PyTorch☆122Updated 2 months ago
- Minimal sharded dataset loaders, decoders, and utils for multi-modal document, image, and text datasets.☆155Updated 9 months ago
- ☆85Updated 11 months ago
- TorchFix - a linter for PyTorch-using code with autofix support☆122Updated 3 weeks ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆105Updated this week
- Minimalistic 4D-parallelism distributed training framework for education purpose☆670Updated this week
- Muon optimizer for neural networks: >30% extra sample efficiency, <3% wallclock overhead☆220Updated 3 weeks ago
- Efficient optimizers☆154Updated this week
- Focused on fast experimentation and simplicity☆65Updated last month
- ☆140Updated 11 months ago
- A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆225Updated last week
- ☆52Updated 2 months ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆240Updated this week
- TensorDict is a pytorch dedicated tensor container.☆868Updated this week
- ☆149Updated 5 months ago