Are there any good gpu linear algebra crates?

• clblast-rs

  0 1 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

  0 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/

• SonarLint

  www.sonarlint.org

  sponsored

  Clean code begins in your IDE with SonarLint. Up your coding game and discover issues early. SonarLint is a free plugin that helps you find & fix bugs and security issues from the moment you start writing code. Install from your favorite IDE marketplace today.

  

• rust-grad

  0 19 0.0 Rust

  Automatic differentiation in Rust with WGPU support

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

• neuronika

  2 959 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