JIT Compilation#

The JIT compilation tools for JAX backend.

  1. Just-In-Time compilation is implemented by the ‘jit()’ function

jit(func[, dyn_vars, static_argnames, ...])

JIT (Just-In-Time) compilation for class objects.