Crate burn_jit

Burn JIT Backend

Re-exports

pub use element::BoolElement;

pub use element::FloatElement;

pub use element::IntElement;

pub use element::JitElement;

pub use cubecl;

Modules

element

Elements for JIT backend

fusionfusion

Module for interacting with fusion

kernel

Kernel module

tensor

Tensor module.

Macros

execute_with_dtype

Macro to execute a kernel/operation for a given element type.

Structs

JitBackend

Generic tensor backend that can be compiled just-in-time to any shader runtime

JitTuneId

ID used to identify a Just-in-Time environment.

Enums

JitAutotuneKey

Key for all autotune-enabled operations

Traits

JitRuntime

Just-in-Time runtime extending the cube runtime.