Skip to content

Experimentation using the xla compiler from rust

License

Notifications You must be signed in to change notification settings

elodin-sys/xla-rs

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

xla-rs

Originally based on https://github.com/LaurentMazare/xla-rs

xla-rs is a Rust wrapper around XLA, a compiler for machine learning and linear algebra. The goal of this project is to create a set of safe bindings close to XLA's C++ API. The intended use of this crate is in other higher-level linear algebra and ML crates. It can be used on its own, but it doesn't promise to be the most ergonomic or well-documented.

Why Fork?

This crate differs from the original xla-rs in a few key ways. The biggest is that the bindings are rewritten to use the cpp and cxx crates. This allows our bindings to be written in line with Rust code, and to have more clear typing. The other large change is that we are using a fork of xla_extension, which allows for building a fully static binary. This fits better the intended use of xla-rs as a low-level crate, by other higher-level libraries.

About

Experimentation using the xla compiler from rust

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Rust 100.0%