Release v1.5.0
Experimental: adds jdc.jit
.
This is the same jax.jit
, but has tweaks for improved typing and autopopulates static_argnums
/static_argnames
based on types annotated with jdc.Static[]
.
Experimental: adds jdc.jit
.
This is the same jax.jit
, but has tweaks for improved typing and autopopulates static_argnums
/static_argnames
based on types annotated with jdc.Static[]
.