Autodiff for Kotlin, compiled not traced. A K2 compiler plugin turns grad {} into gradient code; lowers to StableHLO and runs on GPU via PJRT or IREE.
kotlin machine-learning deep-learning jvm gpu automatic-differentiation cuda tensor k2 autodiff ptx differentiable-programming xla kotlin-compiler-plugin mlir iree llm-inference stablehlo pjrt diffkt
-
Updated
Oct 29, 2026 - Kotlin