thunderkittens ported to rust w/ cuda-oxide.

cuda-oxide is a Rust-to-CUDA compiler that lets you write (SIMT) GPU kernels in safe(ish), idiomatic Rust. It compiles standard Rust code directly to PTX — no DSLs, no foreign language bindings, ju…

cuTile Rust provides a safe, tile-based kernel programming DSL for the Rust programming language. It features a safe host-side API for passing tensors to asynchronously executed kernel functions.

Manage system config using nix on any distro

🤖 Just a command runner

This repository is the official implementation of ZSC-Eval: An Evaluation Toolkit and Benchmark for Multi-agent Zero-shot Coordination. Pre-trained Agent Zoo: https://huggingface.co/Leoxxxxh/ZSC-Ev…

PyTorch Single Controller

Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/

(Crafter + NetHack) in JAX. ICML 2024 Spotlight.

Ditto is an open-source framework that enables direct conversion of HuggingFace PreTrainedModels into TensorRT-LLM engines.

Ecosystem of libraries and tools for writing and executing fast GPU code fully in Rust.

verl/HybridFlow: A Flexible and Efficient RL Post-Training Framework

A collection of robotics simulation environments for reinforcement learning

Fancy extension for std::error::Error with pretty, detailed diagnostic printing.

Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)

An open source re-implementation of Chris Sawyer's Locomotion

Online Goal-Conditioned Reinforcement Learning in JAX. ICLR 2025 Spotlight.

Educational framework exploring ergonomic, lightweight multi-agent orchestration. Managed by OpenAI Solution team.

Computing with Python functions.

A suite of test scenarios for multi-agent reinforcement learning.

Mirror of https://code.tecosaur.net/tec/About.jl

A Git-compatible VCS that is both simple and powerful

Monte Carlo tree search in JAX

The Unity Machine Learning Agents Toolkit (ML-Agents) is an open-source project that enables games and simulations to serve as environments for training intelligent agents using deep reinforcement …

DORA (Dataflow-Oriented Robotic Architecture) is middleware designed to streamline and simplify the creation of AI-based robotic applications. It offers low latency, composable, and distributed dat…

Experimental projects related to TensorRT

You have a GPU. Now make it shareable and accessible from anywhere in the world.

Bend 2: a fast language that blocks AI mistakes via proof. Install: curl -fsSL https://bend-lang.com/install.sh | sh