A Rust Framework for Tracing, Automatic Differentiation, and Just-In-Time Compilation
-
Updated
Apr 15, 2026 - Rust
A Rust Framework for Tracing, Automatic Differentiation, and Just-In-Time Compilation
Experimentation using the xla compiler from rust
Rust-powered transpiler for PyTorch → JAX. Converts stateful FX/ONNX graphs to pure JAX primitives in <1ms. Validated on ResNet-18, DeiT-Small, and Transformers with MSE < 1e-10.
Graph-native numerical computation for Rust, compiled to XLA
Add a description, image, and links to the xla topic page so that developers can more easily learn about it.
To associate your repository with the xla topic, visit your repo's landing page and select "manage topics."