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 3 years ago
Alternatives and similar repositories for stoke
Users that are interested in stoke are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- spock is a framework that helps manage complex parameter configurations during research and development of Python applications☆145Nov 3, 2023Updated 2 years ago
- Ludwig benchmark☆20May 11, 2026Updated 5 months 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
- Utility functions that I reuse across different projects☆15Jun 4, 2021Updated 5 years ago
- A library helping to gather stats and run checks during training deep learning models with Pytorch☆35Mar 6, 2022Updated 4 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- ☆15Feb 28, 2022Updated 4 years ago
- Single Image Texture Translation for Data Augmentation☆61Feb 22, 2023Updated 3 years ago
- Implementation of Uformer, Attention-based Unet, in Pytorch☆97Oct 26, 2021Updated 5 years ago
- Run compute jobs on AWS as if you were running them locally.☆123Dec 12, 2021Updated 4 years ago
- Server wrapper for ml models☆11Sep 11, 2019Updated 7 years ago
- Take control back!☆18Jun 22, 2022Updated 4 years ago
- Utility functions for weights and biases (wandb).☆11Sep 17, 2024Updated 2 years ago
- Pipeline for training LSA models using Scikit-Learn.☆23Oct 12, 2021Updated 5 years ago
- Package speculatively provides a simple mechanism to re-execute a task in parallel only after some initial timeout has elapsed.☆10Jul 11, 2025Updated last year
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- Code accompanying the "Debugging machine learning in production" talk☆31Feb 22, 2022Updated 4 years ago
- Some small utility modules to help with pandas, numpy and sklearn usage in other projects☆22Apr 15, 2015Updated 11 years ago
- A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training☆28Jul 12, 2022Updated 4 years ago
- MABWiser: Contextual Multi-Armed Bandits Library☆297Sep 5, 2024Updated 2 years ago
- Implementation of Kronecker Attention in Pytorch☆20Sep 12, 2020Updated 6 years ago
- Image data augmentation scheduler for albumentations transforms☆19May 1, 2021Updated 5 years ago
- ☆10Mar 31, 2025Updated last year
- Jurity: Fairness & Evaluation Library☆59Oct 11, 2024Updated 2 years ago
- PyTorch dataset extended with map, cache etc. (tensorflow.data like)☆331Jun 13, 2022Updated 4 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Extend MLflow's functionality☆10Dec 11, 2022Updated 3 years ago
- PyTorch implementation of L2L execution algorithm☆109Jan 16, 2023Updated 3 years ago
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,485May 2, 2025Updated last year
- 7th place solution☆22Mar 28, 2021Updated 5 years ago
- [IJCAI'25] Bandit-based Adaptive Large-Neighborhood Search☆24Aug 13, 2026Updated 2 months ago
- Knowledge-Aware machine LEarning (KALE): accessible machine learning from multiple sources for interdisciplinary research, part of the 🔥…☆490Updated this week
- An open-source AutoML Library based on PyTorch☆306Updated this week
- ☁️ Export Ploomber pipelines to Kubernetes (Argo), Airflow, AWS Batch, SLURM, and Kubeflow.☆47Mar 10, 2025Updated last year
- Code for converting FDDB data to the format required by YOLO☆11Apr 14, 2017Updated 9 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- An ML framework to accelerate research and its path to production.☆271Sep 3, 2024Updated 2 years ago
- ☆252Jul 25, 2024Updated 2 years ago
- 🤔 When in Doubt: Improving Classification Performance with Alternating Normalization [Findings of EMNLP2021]☆15Oct 29, 2021Updated 5 years ago
- The official Jupyterlab extension for conda-store☆16Nov 26, 2024Updated last year
- Helps you write algorithms in PyTorch that adapt to the available (CUDA) memory☆434Aug 29, 2024Updated 2 years ago
- PyTorch extensions for high performance and large scale training.☆3,404Apr 26, 2025Updated last year
- A Python package of computer vision models for the Equinox ecosystem.☆113Jul 19, 2024Updated 2 years ago