You are viewing the development documentation (main). Switch to the stable release docs.

jax_function_override¶

probly.representation.sample.jax_functions.jax_function_override(jax_func: _JaxBoundFunction) → _JaxFunction[source]¶: Decorator to convert a bound jax function to a jax function.