kd.random

[[Source]]

Small wrapper around jax.random for OO API.

Symbols

Class

kd.random.PRNGKey	Small wrapper around jax.random key arrays to reduce boilerplate.