Are there any good gpu linear algebra crates?

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

Our great sponsors
  • SonarLint - Clean code begins in your IDE with SonarLint
  • InfluxDB - Access the most powerful time series database as a service
  • SaaSHub - Software Alternatives and Reviews
  • 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.

  • 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:

  • SonarLint

    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

    Automatic differentiation in Rust with WGPU support

    An example of it's usage:

  • 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