Easily run PyTorch on multiple GPUs & machines
☆59Jan 8, 2026Updated last month
Alternatives and similar repositories for torchrunx
Users that are interested in torchrunx are comparing it to the libraries listed below
Sorting:
- An implementation of the Llama architecture, to instruct and delight☆21May 31, 2025Updated 9 months ago
- ☆48Aug 29, 2024Updated last year
- PyTorch implementation for all methods and environments in the paper "MIMEx: Intrinsic Rewards from Masked Input Modeling"☆16May 17, 2023Updated 2 years ago
- Python tools☆14Oct 22, 2023Updated 2 years ago
- A Flexible Toolkit for Dense Retrieval☆44Nov 12, 2025Updated 3 months ago
- RL environments and PPO training code to develop trajectory tracking controllers for various robot systems, written in Jax.☆19May 20, 2025Updated 9 months ago
- Experiments on GPT-3's ability to fit numerical models in-context.☆14Aug 11, 2022Updated 3 years ago
- Minimalistic, hackable PyTorch implementation of SimSiam in ~400 lines. Achieves good performance on ImageNet with ResNet50. Features dis…☆21Nov 25, 2024Updated last year
- ☆18May 19, 2023Updated 2 years ago
- The official implementation of DMEL the method presented in the paper "DMEL: The differentiable log-Mel spectrogram as a trainable layer …☆22Dec 21, 2024Updated last year
- ☆17Jun 19, 2023Updated 2 years ago
- ☆20Jul 12, 2023Updated 2 years ago
- Synthesise image from semantic map using Cascaded Refinement Netowork and PyTorch.☆19Apr 18, 2018Updated 7 years ago
- applying audio FX with text descriptors☆33Nov 12, 2025Updated 3 months ago
- Utilities for Training Very Large Models☆58Sep 25, 2024Updated last year
- Flow Policy Gradients for Robot Control☆149Feb 4, 2026Updated last month
- Cray-LM unified training and inference stack.☆22Jan 30, 2025Updated last year
- Multi-agent simulator in Jax for research and teaching in AI & ALife☆31Updated this week
- Robot + mesh spherisation tool☆68Jan 16, 2026Updated last month
- NanoGPT-speedrunning for the poor T4 enjoyers☆73Apr 22, 2025Updated 10 months ago
- ICASSP 2022☆61Oct 12, 2021Updated 4 years ago
- mlx implementations of various transformers, speedups, training☆33Dec 14, 2023Updated 2 years ago
- Jax like function transformation engine but micro, microjax☆34Oct 25, 2024Updated last year
- Minimal but scalable implementation of large language models in JAX☆35Nov 28, 2025Updated 3 months ago
- Clean RL implementation using MLX☆35Mar 8, 2024Updated last year
- Automatically take good care of your preemptible TPUs☆37May 15, 2023Updated 2 years ago
- experiments about AudioSet☆43Jul 22, 2023Updated 2 years ago
- A subset of PyTorch's neural network modules, written in Python using OpenAI's Triton.☆595Aug 12, 2025Updated 6 months ago
- chaii: hindi and tamil question answering☆35Dec 5, 2021Updated 4 years ago
- git tracking for python notebooks☆12Jun 15, 2017Updated 8 years ago
- A Redis-compatible in-memory database server written in Rust with MLua-based Lua 5.1 scripting☆17Nov 28, 2025Updated 3 months ago
- An opinionated NLP research template☆10Aug 29, 2024Updated last year
- Implementaion of Generic L-layer Neural Network from Scratch☆12May 14, 2018Updated 7 years ago
- A Library for Scaling Mixed-Integer Optimization-Based Machine Learning.☆12Jun 24, 2024Updated last year
- ConfigParser class with AES-256 symmetric encryption support☆10Mar 8, 2024Updated last year
- ✍️ A browser add-on (Firefox, Chrome, Thunderbird) that allows you to autocorrect common text sequences and convert text characters to a …☆12Feb 9, 2026Updated 3 weeks ago
- Description of TAIL website☆14May 8, 2024Updated last year
- Mixture of Experts from scratch☆13Apr 12, 2024Updated last year
- Triton-based Symmetric Memory operators and examples☆86Jan 15, 2026Updated last month