ML frameworks configuration

Configurations for behavior modification of Machine Learning (ML) frameworks.

---

JAX

JAX specific configuration

API

skrl.config.jax.backend: str = "numpy"

Backend used by the different components to operate and generate arrays

This configuration excludes models and optimizers. Supported backend are: "numpy" and "jax"

skrl.config.jax.key: jnp.ndarray = [0, 0]

Pseudo-random number generator (PRNG) key