A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-precision, and PyTorch extensions.
☆66Jul 31, 2023Updated 2 years ago
Alternatives and similar repositories for stoke
Users that are interested in stoke are comparing it to the libraries listed below
Sorting:
- spock is a framework that helps manage complex parameter configurations during research and development of Python applications☆142Nov 3, 2023Updated 2 years ago
- Distributed ML Optimizer☆35Jul 28, 2021Updated 4 years ago
- Ludwig benchmark☆19Mar 13, 2022Updated 4 years ago
- [AMAI 2024] Selective: Feature Selection Library☆69Sep 9, 2025Updated 6 months ago
- A library helping to gather stats and run checks during training deep learning models with Pytorch☆35Mar 6, 2022Updated 4 years ago
- FuturePool is a package that introduce known concept of multiprocessing Pool to the async/await world. It allows for easy translation fro…☆14Nov 10, 2024Updated last year
- Functional deep learning☆107Nov 23, 2022Updated 3 years ago
- Single Cell Multiplexed Imaging Jupyter Voila Dashboard using CODEX Data☆17Jul 6, 2023Updated 2 years ago
- PyTorch Lightning + Hydra + MLFlow - Deep learning template☆24Sep 14, 2021Updated 4 years ago
- ☆126Jun 3, 2024Updated last year
- Package speculatively provides a simple mechanism to re-execute a task in parallel only after some initial timeout has elapsed.☆10Jul 11, 2025Updated 8 months ago
- [IJAIT 2021] MABWiser: Contextual Multi-Armed Bandits Library☆279Sep 5, 2024Updated last year
- A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training☆28Jul 12, 2022Updated 3 years ago
- Some small utility modules to help with pandas, numpy and sklearn usage in other projects☆22Apr 15, 2015Updated 10 years ago
- Utility functions for weights and biases (wandb).☆11Sep 17, 2024Updated last year
- Image data augmentation scheduler for albumentations transforms☆19May 1, 2021Updated 4 years ago
- Large dataset storage format for Pytorch☆45Aug 19, 2021Updated 4 years ago
- Dataset corresponding to the paper: "Form2Seq : A Framework for Higher-Order Form Structure Extraction"☆10Feb 17, 2021Updated 5 years ago
- a lightweight transformer library for PyTorch☆71Nov 2, 2021Updated 4 years ago
- Dynamic config system based on python classes☆12Jan 27, 2023Updated 3 years ago
- [ACM 2024] Jurity: Fairness & Evaluation Library☆57Oct 11, 2024Updated last year
- An open source implementation of CLIP.☆33Nov 7, 2022Updated 3 years ago
- 7th place solution☆22Mar 28, 2021Updated 4 years ago
- Experiment Manager☆22Jan 15, 2020Updated 6 years ago
- ☁️ Export Ploomber pipelines to Kubernetes (Argo), Airflow, AWS Batch, SLURM, and Kubeflow.☆45Mar 10, 2025Updated last year
- Code for fitting masks to face images in the wild☆10Jul 29, 2021Updated 4 years ago
- A minimal PyTorch Lightning OpenAI GPT w DeepSpeed Training!☆112May 11, 2023Updated 2 years ago
- A PyTorch implementation of computing mean average precision in parallel☆16Jul 7, 2022Updated 3 years ago
- A library to inspect and extract intermediate layers of PyTorch models.☆473May 12, 2022Updated 3 years ago
- Template project for performing inference on ONNX models in Python.☆17Jul 2, 2022Updated 3 years ago
- The official Jupyterlab extension for conda-store☆16Nov 26, 2024Updated last year
- Code for converting FDDB data to the format required by YOLO☆11Apr 14, 2017Updated 8 years ago
- Pytorch based library to rank predicted bounding boxes using text/image user's prompts.☆52Dec 25, 2021Updated 4 years ago
- ☆17Feb 16, 2024Updated 2 years ago
- A PyTorch implementation of Neighbourhood Components Analysis.☆399Sep 12, 2020Updated 5 years ago
- A library to instantiate any Python object from configuration files.☆24Oct 12, 2022Updated 3 years ago
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,475May 2, 2025Updated 10 months ago
- ☆16Jul 26, 2020Updated 5 years ago
- ☆153May 25, 2020Updated 5 years ago