Are there any good gpu linear algebra crates?

This page summarizes the projects mentioned and recommended in the original post on reddit.com/r/rust

Our great sponsors
  • OPS - Build and Run Open Source Unikernels
  • SonarLint - Deliver Cleaner and Safer Code - Right in Your IDE of Choice!
  • Scout APM - Less time debugging, more time building
  • GitHub repo clblast-rs

    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.

  • GitHub repo ndarray

    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/

  • OPS

    OPS - Build and Run Open Source Unikernels. Quickly and easily build and deploy open source unikernels in tens of seconds. Deploy in any language to any cloud.

  • GitHub repo rust-grad

    Automatic differentiation in Rust with WGPU support

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

  • GitHub repo neuronika

    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!

NOTE: The number of mentions on this list indicates mentions on common posts plus user suggested alternatives. Hence, a higher number means a more popular project.

Suggest a related project

Related posts