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

jax_sample_internals¶

probly.representation.sample.jax_functions.jax_sample_internals(_: object) → JaxSampleInternals | None¶
probly.representation.sample.jax_functions.jax_sample_internals(sample: JaxSample) → JaxSampleInternals[Array]: Get internals for a sample array.