RAI: Rust ML framework with composable transformations like JAX.
☆95Jul 19, 2024Updated 2 years ago
Alternatives and similar repositories for rai
Users that are interested in rai are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Rustic bindings to the IREE Compiler/Runtime☆30Updated this week
- Experimentation using the xla compiler from rust☆117Updated this week
- Save pytrees efficiently in hdf5 files.☆15Updated this week
- Rustic bindings for IREE☆18Feb 18, 2023Updated 3 years ago
- a GGUF file parser☆25Aug 31, 2026Updated 2 months ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- ☆12Feb 10, 2026Updated 8 months ago
- A metrics library for the JAX ecosystem☆40Mar 16, 2023Updated 3 years ago
- Graph model execution API for Candle☆19Jul 27, 2025Updated last year
- ☆14Nov 27, 2025Updated 11 months ago
- A compiler written in Mojo 🔥 and generates RISC-V assembly☆17Jun 25, 2024Updated 2 years ago
- Sequential decision-making in JAX.☆10Sep 21, 2026Updated last month
- Classes/structs, pattern matching, safe counters... and more! Your one-stop library for programming tools not already in core Typst.☆30Oct 16, 2026Updated 2 weeks ago
- Blazingly Fast Pseudo Random Number Generator written in Rust☆22Sep 21, 2024Updated 2 years ago
- A high-performance reinforcement learning library in jax specialized for robotic learning☆22Sep 4, 2023Updated 3 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Rust library for scheduling, managing resources, and running DAGs 🌙☆40Jul 31, 2026Updated 3 months ago
- Portable and lightweight brain segmentation in the terminal!☆23Oct 19, 2026Updated last week
- Game Engine From Scratch -- Rust China Conference 2020 topic by LemonHX and his team.☆14Dec 16, 2020Updated 5 years ago
- Tensor Compute Primitives: Mid-level Intermediate Representation for Machine Learning Programs☆35Jan 30, 2025Updated last year
- A Collection of Papers & Notes in Programming Language & Formal Verification☆17May 10, 2022Updated 4 years ago
- JMP is a Mixed Precision library for JAX.☆214Oct 5, 2026Updated 3 weeks ago
- For when you really, really just want to know that your config changed☆14Jun 17, 2023Updated 3 years ago
- Deep learning in Rust, with shape checked tensors and neural networks☆1,939Jul 23, 2024Updated 2 years ago
- Library for reading and processing ML training data.☆800Updated this week
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- ☆658Oct 20, 2026Updated last week
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆327Oct 14, 2026Updated 2 weeks ago
- My code repo for random-walks.org☆16Feb 1, 2026Updated 8 months ago
- Your one stop CLI for ONNX model analysis.☆48Nov 13, 2022Updated 3 years ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆46Jul 19, 2025Updated last year
- General matrix multiplication of f32 and f64 matrices in Rust. Supports matrices with general strides.☆273Jul 14, 2026Updated 3 months ago
- JAX implementation of RL algorithms and vectorized environments☆50Dec 26, 2023Updated 2 years ago
- A minimal OpenCL, CUDA, Vulkan and host CPU array manipulation engine / framework.☆78Jul 2, 2026Updated 3 months ago
- ☆12Aug 19, 2021Updated 5 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.
- Tidy autoregressive inference in JAX☆15Sep 1, 2025Updated last year
- ☆28Aug 10, 2023Updated 3 years ago
- Macros enabling parallelism using the rayon crate☆21Feb 10, 2024Updated 2 years ago
- Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.☆137Updated this week
- Codes for "Efficient Offline Policy Optimization with a Learned Model", ICLR2023☆31Jul 18, 2023Updated 3 years ago
- Inference at the speed of light.☆3,036Updated this week
- A Rust Framework for Tracing, Automatic Differentiation, and Just-In-Time Compilation☆19Updated this week