Are there any good gpu linear algebra crates?

• clblast-rs

  1 3 0.0 Rust

  clblast bindings for rust

I created clblast-rs, which binds to the clblast library. Main reason to use it is if you want something relatively fast that also allows you to write your own kernels to operate on the same buffers, that clblast uses.

• ndarray

  2 6 0.0

  ndarray: an N-dimensional array with array views, multidimensional slicing, and efficient operations (by RustyBamboo)

Although very primitive (only with element-wise operations and dot product operations), and probably would not fit your criteria of "good" crate, you can checkout my experimental fork of the ndarray crate with WGPU support: https://github.com/RustyBamboo/ndarray/

• WorkOS

  workos.com sponsored

  The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.

  

• rust-grad

  2 21 0.0 Rust

  Automatic differentiation in Rust with WGPU support

An example of it's usage: https://github.com/RustyBamboo/rust-grad

• neuronika

  19 1,033 0.0 Rust

  Tensors and dynamic neural networks in pure Rust.

  

Interesting! I'm also working on an autograd framework based on ndarray, it would be cool, if you are interested or have time, to try and experiment with GPUs!

Suggest a related project