pytorch / torchevalLinks
A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to facilitate metric computation in distributed training and tools for PyTorch model evaluations.
☆235Updated 6 months ago
Alternatives and similar repositories for torcheval
Users that are interested in torcheval are comparing it to the libraries listed below
Sorting:
- Configuration classes enabling type-safe PyTorch configuration for Hydra apps☆219Updated 2 years ago
- TensorDict is a pytorch dedicated tensor container.☆944Updated this week
- Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory☆437Updated 10 months ago
- A library to inspect and extract intermediate layers of PyTorch models.☆473Updated 3 years ago
- TorchFix - a linter for PyTorch-using code with autofix support☆143Updated 5 months ago
- Named tensors with first-class dimensions for PyTorch☆332Updated 2 years ago
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆252Updated 2 years ago
- A lightweight library designed to accelerate the process of training PyTorch models by providing a minimal, but extensible training loop …☆187Updated last month
- ☆349Updated 2 weeks ago
- Implementation of a Transformer, but completely in Triton☆270Updated 3 years ago
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆214Updated 2 years ago
- The merlin dataloader lets you rapidly load tabular data for training deep leaning models with TensorFlow, PyTorch or JAX☆420Updated last year
- For optimization algorithm research and development.☆522Updated this week
- TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and sup…☆373Updated this week
- Lightning HPO & Training Studio App☆18Updated 2 years ago
- optimizer & lr scheduler & loss function collections in PyTorch☆314Updated this week
- Implementation of a memory efficient multi-head attention as proposed in the paper, "Self-attention Does Not Need O(n²) Memory"☆379Updated 2 years ago
- Common Python utilities and GitHub Actions in Lightning Ecosystem☆56Updated 2 weeks ago
- Pytorch Lightning Distributed Accelerators using Ray☆213Updated last year
- Transform datasets at scale. Optimize datasets for fast AI model training.☆509Updated last week
- Demystify RAM Usage in Multi-Process Data Loaders☆194Updated 2 years ago
- Memory Efficient Attention (O(sqrt(n)) for Jax and PyTorch☆184Updated 2 years ago
- The spiritual successor to knockknock for PyTorch Lightning, get notified when your training ends☆77Updated 7 months ago
- Recipes are a standard, well supported set of blueprints for machine learning engineers to rapidly train models using the latest research…☆323Updated last week
- [Prototype] Tools for the concurrent manipulation of variably sized Tensors.☆251Updated 2 years ago
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆208Updated last year
- A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.☆1,213Updated this week
- Scalable and Performant Data Loading☆288Updated this week
- Code release for "Dropout Reduces Underfitting"☆313Updated 2 years ago
- Memory mapped numpy arrays of varying shapes☆299Updated last year