Expand description
JIT Compiler for GPU Kernels
Compiles operations at runtime for maximum performance This is the secret weapon to beat JAX!
Structs§
- Compiled
Kernel - JIT-compiled kernel
- Graph
Signature - Graph signature for caching
- JitCompiler
- JIT compiler that generates and caches optimized kernels
- Kernel
Launcher - Optimized kernel launcher