Rust-CUDA VS Enzyme

Compare Rust-CUDA vs Enzyme and see what are their differences.

Our great sponsors
  • Scout APM - Less time debugging, more time building
  • talent.io - Download talent.io’s Tech Salary Report
  • SonarLint - Clean code begins in your IDE with SonarLint
Rust-CUDA Enzyme
23 12
1,879 781
5.3% 2.6%
8.7 9.5
7 days ago 3 days ago
Rust LLVM
Apache License 2.0 GNU General Public License v3.0 or later
The number of mentions indicates the total number of mentions that we've tracked plus the number of user suggested alternatives.
Stars - the number of stars that a project has on GitHub. Growth - month over month growth in stars.
Activity is a relative number indicating how actively a project is being developed. Recent commits have higher weight than older ones.
For example, an activity of 9.0 indicates that a project is amongst the top 10% of the most actively developed projects that we are tracking.

Rust-CUDA

Posts with mentions or reviews of Rust-CUDA. We have used some of these posts to build our list of alternatives and similar projects. The last one was on 2022-09-15.
  • Can Rust do Computer Graphics and ML?
    5 projects | reddit.com/r/rust | 15 Sep 2022
    There’s also https://github.com/Rust-GPU/Rust-CUDA
  • There is framework for everything.
    107 projects | reddit.com/r/ProgrammerHumor | 4 Aug 2022
  • C++ is making me depressed / CUDA question
    7 projects | reddit.com/r/rust | 20 Jul 2022
    And here's an example on how to add two floats using Rust-CUDA: https://github.com/Rust-GPU/Rust-CUDA/blob/master/examples/cuda/gpu/add_gpu/src/lib.rs
    7 projects | reddit.com/r/rust | 20 Jul 2022
    I can't say whether or not it works well, and seems maybe fiddly to install, but Rust-CUDA might work for you.
  • Does Rust still depend on LLVM?
    3 projects | reddit.com/r/rust | 18 May 2022
    Kinda - SPIR-V is an intermediate representation that can be consumed by graphics drivers and compiled down by them to the GPU's native code. In that sense it's more similar to emitting LLVM IR than e.g. x86 machine code. Similarly, Rust-CUDA emits NVVM IR, which is a subset of LLVM IR.
  • State of GPGPU in 2022
    4 projects | reddit.com/r/u_KestM | 27 Apr 2022
    Rust-GPU look promising, also Rust-CUDA, also I see OpenCL 3.0 wrapper
  • Silverblue loads nouveau instead of installed nvidia
    3 projects | reddit.com/r/silverblue | 12 Apr 2022
    rpm-ostree install https://download1.rpmfusion.org/free/fedora/rpmfusion-free-release-$(rpm -E %fedora).noarch.rpm https://download1.rpmfusion.org/nonfree/fedora/rpmfusion-nonfree-release-$(rpm -E %fedora).noarch.rpm sudo rpm-ostree uninstall akmod-nvidia sudo rpm-ostree install akmod-nvidia-470xx rpm-ostree kargs --append=rd.driver.blacklist=nouveau --append=modprobe.blacklist=nouveau --append=nvidia-drm.modeset=1 sudo systemctl reboot sudo rpm-ostree install nvidia-settings-470xx xorg-x11-drv-nvidia-470xx-power sudo rpm-ostree uninstall xorg-x11-drv-nvidia-cuda sudo rpm-ostree install xorg-x11-drv-nvidia-470xx-cuda sudo rpm-ostree install xorg-x11-drv-nvidia-470xx-devel sudo rpm-ostree install xorg-x11-drv-nvidia-470xx-cuda-devel sudo rpm-ostree install cuda-samples sudo rpm-ostree install vulkan-tools vkmark mesa-vulkan-devel sudo rpm-ostree install libshaderc-devel sudo rpm-ostree install clang clang-tools-extra libstdc++-devel sudo rpm-ostree install glib2-devel glib-devel avahi-gobject-devel sudo rpm-ostree install cairo-devel pango-devel gdk-pixbuf2-devel sudo rpm-ostree install graphene-devel gtk4-devel cairo-gobject-devel sudo systemctl reboot modinfo /usr/lib/modules/$(uname -r)/kernel/drivers/video/nvidia.ko | grep ^version find /usr/lib/modules -name nvidia.ko -exec modinfo {} \; sudo lspci -v | grep -A 20 VGA git clone https://github.com/Rust-GPU/Rust-CUDA.git git clone https://github.com/vulkano-rs/vulkano.git git clone https://github.com/Relm4/relm4.git glxgears glxinfo glxgears glxinfo vkcube vkcubepp ./teapot ./triangle ./occlusion-query ./interactive_fractal
  • [P] C++ Machine Learning Library Built From Scratch by a 16-Year-Old High Schooler
    3 projects | reddit.com/r/MachineLearning | 12 Feb 2022
  • Writing Vulkan SPIR-V shaders in C++?
    4 projects | reddit.com/r/vulkan | 12 Feb 2022
    Is Rust CUDA, "writing [...] GPU code fully in Rust" (from its "about" description), but still early in development, relevant?
  • Announcing Rust-CUDA 0.3; Hardware raytracing, cuBLAS, cuDNN, atomics, and more
    3 projects | reddit.com/r/rust | 7 Feb 2022

Enzyme

Posts with mentions or reviews of Enzyme. We have used some of these posts to build our list of alternatives and similar projects. The last one was on 2022-06-08.
  • What is a really cool thing you would want to write in Rust but don't have enough time, energy or bravery for?
    21 projects | reddit.com/r/rust | 8 Jun 2022
    Have you taken a look at enzymeAD? There is a group porting it to rust.
  • The Julia language has a number of correctness flaws
    19 projects | news.ycombinator.com | 16 May 2022
    Enzyme dev here, so take everything I say as being a bit biased:

    While, by design Enzyme is able to run very fast by operating within the compiler (see https://proceedings.neurips.cc/paper/2020/file/9332c513ef44b... for details) -- it aggressively prioritizes correctness. Of course that doesn't mean that there aren't bugs (we're only human and its a large codebase [https://github.com/EnzymeAD/Enzyme], especially if you're trying out newly-added features).

    Notably, this is where the current rough edges for Julia users are -- Enzyme will throw an error saying it couldn't prove correctness, rather than running (there is a flag for "making a best guess, but that's off by default"). The exception to this is garbage collection, for which you can either run a static analysis, or stick to the "officially supported" subset of Julia that Enzyme specifies.

    Incidentally, this is also where being a cross-language tool is really nice -- namely we can see edge cases/bug reports from any LLVM-based language (C/C++, Fortran, Swift, Rust, Python, Julia, etc). So far the biggest code we've handled (and verified correctness for) was O(1million) lines of LLVM from some C++ template hell.

    I will also add that while I absolutely love (and will do everything I can to support) Enzyme being used throughout arbitrary Julia code: in addition to exposing a nice user-facing interface for custom rules in the Enzyme Julia bindings like Chris mentioned, some Julia-specific features (such as full garbage collection support) also need handling in Enzyme.jl, before Enzyme can be considered an "all Julia AD" framework. We are of course working on all of these things (and the more the merrier), but there's only a finite amount of time in the day. [^]

    [^] Incidentally, this is in contrast to say C++/Fortran/Swift/etc, where Enzyme has much closer to whole-language coverage than Julia -- this isn't anything against GC/Julia/etc, but we just have things on our todo list.

  • Jax vs. Julia (Vs PyTorch)
    4 projects | news.ycombinator.com | 4 May 2022
    Idk, Enzyme is pretty next gen, all the way down to LLVM code.

    https://github.com/EnzymeAD/Enzyme

  • What's everyone working on this week (7/2022)?
    15 projects | reddit.com/r/rust | 14 Feb 2022
    I'm working on merging my build-tool for (oxide)-enzyme into Enzyme itself. Also looking into improving the documentation.
  • Trade-Offs in Automatic Differentiation: TensorFlow, PyTorch, Jax, and Julia
    7 projects | news.ycombinator.com | 25 Dec 2021
    that seems one of the points of enzyme[1], which was mentioned in the article.

    [1] - https://enzyme.mit.edu/

    being able in effect do interprocedural cross language analysis seems awesome.

  • Enzyme: towards state-of-the-art AutoDiff in Rust
    2 projects | reddit.com/r/rust | 11 Dec 2021
    Enzyme is an LLVM (incubator) project, which performs automatic differentiation of LLVM-IR code. Here is an introduction to AutoDiff, which was recommended by @DoogoMiercoles in an earlier post. You can also try it online, if you know some C/C++: https://enzyme.mit.edu/explorer.
  • Oxide-Enzyme: Integrating LLVM's Static Automatic Differentiation Plugin
    3 projects | reddit.com/r/rust | 30 Nov 2021
    To give a little bit of context here, this is a Rust frontend for Enzyme itself, which is a leading Auto-Diff tool. The key advantage is that unlike most of the existing tools it does generate gradient functions after applying a lot of (LLVM's) optimizations, which leads to very efficient gradients (benchmarks here: https://enzyme.mit.edu/). Working on LLVM level also allows it to work across language barriers. Finally it is also the first AD library to support generic AMD-HIP / NVIDIA-CUDA code and works also with OpenMP and MPI. https://c.wsmoses.com/papers/EnzymeGPU.pdf I have intentions to add rayon support, since that is more likely to be used on our Rust side :)
  • Cerebras’ New Monster AI Chip Adds 1.4T Transistors
    4 projects | news.ycombinator.com | 22 Apr 2021
    The answer is an API, like NNAPI. AD is a frontend concern and doesn't really matter to accelerators.

    For AD, I am bullish for Enzyme, which does AD on LLVM IR, avoiding deep compiler integration: https://enzyme.mit.edu/

  • Linfa has a website now!
    4 projects | reddit.com/r/rust | 8 Mar 2021
    this is definitely on our agenda, there are several autograd libraries on language level, which you can find on crates.io. The real game changer would be automatic differentiation of LLVM IR code, because we could then construct any order derivative without a special language construct. The Enzyme project provides the faculties and we currently trying to figure out how to integrate it into the ecosystem
  • PyTorch 1.8 release with AMD ROCm support
    8 projects | news.ycombinator.com | 4 Mar 2021
    Have you tried using Enzyme[0] on Numba IR?

    [0] https://enzyme.mit.edu

What are some alternatives?

When comparing Rust-CUDA and Enzyme you can also consider the following projects:

Zygote.jl - 21st century AD

wgpu - Safe and portable GPU abstraction in Rust, implementing WebGPU API.

rust-gpu - 🐉 Making Rust a first-class language and ecosystem for GPU shaders 🚧

rust-ndarray - ndarray: an N-dimensional array with array views, multidimensional slicing, and efficient operations

Flux.jl - Relax! Flux is the ML library that doesn't make you tensor

Pytorch - Tensors and Dynamic neural networks in Python with strong GPU acceleration

CUDA.jl - CUDA programming in Julia.

faust - Functional programming language for signal processing and sound synthesis

linfa - A Rust machine learning framework.

tensorflow - An Open Source Machine Learning Framework for Everyone

tensorflow_macos - TensorFlow for macOS 11.0+ accelerated using Apple's ML Compute framework.

Seed - A Rust framework for creating web apps