Reactant.jl

Appearance

0.3k

Reactant.jl DocsOptimizing Julia Functions with MLIR

Optimize Julia Functions With MLIR and XLA for High-Performance Execution on CPU, GPU, TPU and more.

Tutorials
API Reference 📚
View on GitHub
Reactant.jl
🚀

Fast & Device Agnostic

Effortlessly execute your code on CPU, GPU, and TPU with MLIR and XLA.

∂

Built-In MLIR AD

Leverage Enzyme-Powered Automatic Differentiation to Differentiate MLIR Functions

🧩

Composable

Executes and optimizes generic Julia code without requiring special rewriting

⚡

Compiler Optimizations

Fancy MLIR Optimizations seamlessly optimize your Julia code

How to Install Reactant.jl? ​

Its easy to install Reactant.jl. Since Reactant.jl is registered in the Julia General registry, you can simply run the following command in the Julia REPL:

julia
julia> using Pkg
julia> Pkg.add("Reactant")

If you want to use the latest unreleased version of Reactant.jl, you can run the following command:

julia
julia> using Pkg
julia> Pkg.add(url="https://github.com/EnzymeAD/Reactant.jl")

Select an Accelerator Backend ​

julia
using Reactant
Reactant.set_default_backend("cpu")
julia
using Reactant
# Set backend to use a GPU if available
Reactant.set_default_backend("gpu")
# Set backend to specifically a CUDA GPU
# Reactant.set_default_backend("cuda")
julia
using Reactant
Reactant.set_default_backend("gpu")
# Set backend to specifically an AMD GPU
# Reactant.set_default_backend("rocm")
julia
using Reactant
Reactant.set_default_backend("tpu")
julia
using Reactant
Reactant.set_default_backend("tt")