Crate burn_jit

Burn JIT Backend

Re-exports

• pub use element::FloatElement;
• pub use element::IntElement;
• pub use element::JitElement;
• pub use cubecl;

Modules

• element
  Elements for JIT backend
• kernel
  Kernel module
• tensor
  Tensor module.

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.
• PrecisionBridge
  Handle precision conversion for the jit backend.

Enums

• JitAutotuneKey
  Key for all autotune-enabled operations

Traits

• JitRuntime
  Just-in-Time runtime extending the cube runtime.