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.
☆67Updated last year
Alternatives and similar repositories for stoke:
Users that are interested in stoke are comparing it to the libraries listed below
- Functional deep learning☆108Updated 2 years ago
- Torch Distributed Experimental☆115Updated 9 months ago
- a lightweight transformer library for PyTorch☆71Updated 3 years ago
- Automate issue discovery for your projects against Lightning nightly and releases.☆47Updated last week
- Simple tooling for marking deprecated functions or classes and re-routing to the new successors' instance.☆51Updated 3 weeks ago
- A collection of Models, Datasets, DataModules, Callbacks, Metrics, Losses and Loggers to better integrate pytorch-lightning with transfor…☆47Updated last year
- pytest plugin for a better developer experience when working with the PyTorch test suite☆44Updated 3 years ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated last year
- An active learning library for Pytorch based on Lightning-Fabric.☆79Updated last year
- Pytorch Lightning seed project with hydra☆18Updated 4 years ago
- ☆25Updated 4 years ago
- Yet Another Neural Network Library 🤔☆27Updated last month
- This repository contains example code to build models on TPUs☆30Updated 2 years ago
- Aloception is a set of package for computer vision: aloscene, alodataset, alonet.☆92Updated last week
- Large dataset storage format for Pytorch☆45Updated 3 years ago
- Placeholder for the opensource Grid AI components☆44Updated 2 years ago
- A case study of efficient training of large language models using commodity hardware.☆69Updated 2 years ago
- A collection of optimizers, some arcane others well known, for Flax.☆29Updated 3 years ago
- Distributed skorch on Ray Train☆57Updated 2 years ago
- PyTorch implementation of L2L execution algorithm☆107Updated 2 years ago
- Productionize machine learning predictions, with ONNX or without☆65Updated last year
- 🛠️ Tools for Transformers compression using PyTorch Lightning ⚡☆83Updated 5 months ago
- GitHub Action That Retrieves Model Runs From Weights & Biases☆60Updated 3 years ago
- Triton Server Component for lightning.ai☆14Updated 2 years ago
- Public repository for managing Grid Platform documentation synced with gitbook on docs.grid.ai☆20Updated 2 years ago
- An Easy To Use PyTorch Computer Vision Library☆51Updated last year
- In-place debugger for the fastai library and pytorch☆28Updated 3 years ago
- Squirrel dataset hub☆42Updated last year
- ☆30Updated 3 years ago
- Distributed ML Optimizer☆32Updated 3 years ago