fidelity / stoke
A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-precision, and PyTorch extensions.
☆66Updated last year
Related projects ⓘ
Alternatives and complementary repositories for stoke
- Productionize machine learning predictions, with ONNX or without☆66Updated 9 months ago
- a lightweight transformer library for PyTorch☆72Updated 3 years ago
- pytest plugin for a better developer experience when working with the PyTorch test suite☆43Updated 2 years ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated last year
- A collection of Models, Datasets, DataModules, Callbacks, Metrics, Losses and Loggers to better integrate pytorch-lightning with transfor…☆47Updated last year
- Torch Distributed Experimental☆116Updated 3 months ago
- Functional deep learning☆106Updated last year
- Pytorch Lightning seed project with hydra☆18Updated 4 years ago
- Test pytorch code with minimal computational overhead☆25Updated last year
- Placeholder for the opensource Grid AI components☆44Updated 2 years ago
- ☆25Updated 3 years ago
- Lightning HPO & Training Studio App☆18Updated last year
- Large dataset storage format for Pytorch☆45Updated 3 years ago
- Automate issue discovery for your projects against Lightning nightly and releases.☆45Updated this week
- An Easy To Use PyTorch Computer Vision Library☆51Updated last year
- This repository contains example code to build models on TPUs☆30Updated last year
- Simple tooling for marking deprecated functions or classes and re-routing to the new successors' instance.☆50Updated 3 weeks ago
- Cyclemoid implementation for PyTorch☆87Updated 2 years ago
- TorchFix - a linter for PyTorch-using code with autofix support☆98Updated last month
- 🛠️ Tools for Transformers compression using PyTorch Lightning ⚡☆79Updated this week
- Distributed ML Optimizer☆31Updated 3 years ago
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆146Updated this week
- Yet Another Neural Network Library 🤔☆25Updated 7 months ago
- PyTorch implementation of L2L execution algorithm☆106Updated last year
- An open source implementation of CLIP.☆32Updated 2 years ago
- Public repository for managing Grid Platform documentation synced with gitbook on docs.grid.ai☆20Updated 2 years ago
- ☆18Updated last year
- An active learning library for Pytorch based on Lightning-Fabric.☆79Updated 6 months ago
- Multidimensional indexing for tensors☆112Updated last year