2 unstable releases

new 0.1.2+cb8ea1b Feb 6, 2025
0.0.0 Jan 30, 2025

#107 in Graphics APIs

Download history 105/week @ 2025-01-27 190/week @ 2025-02-03

295 downloads per month

Apache-2.0

290KB
4.5K SLoC

triton-distributed

Overview

This repository contains the core components of a distributed GenAI inference framework written in Rust.

Features:

  • A High level API defined in component.rs to build distributed applications.
  • A Distributed runtime to manage the distributed execution of the inference graph.
  • Uses NATS for component communication and etcd for service discovery, allowing engines to be distributed across multiple nodes while maintaining a unified processing graph.
  • Modular design makes it easy to build inference pipelines by composing reusable components.

Install Rust:

curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh

Rust Installation Guide

Rust Build

cargo build

Run Tests

cargo test

Dependencies

~32–47MB
~894K SLoC