JaxSampleCreator¶ class probly.representation.sample.jax_functions.JaxSampleCreator(*args, **kwargs)[source]¶ Bases: Protocol, Generic Protocol for creating sample arrays. __call__(array: D, sample_axis: int, weights: jax.Array | None) → Any[source]¶ Create a sample array from a JAX array and a sample axis.