rlmesh.jax
Experimental JAX-backed RLMesh clients and tensor helpers.
Experimental
import rlmesh.jaxClasses
- EnvFactoryExperimentalJAX-backed
EnvFactory: served envs speak JAX arrays. - ModelExperimentalExperimental JAX-backed model:
predictworks in JAX values. - RemoteEnvExperimentalExperimental JAX-backed remote client for one environment.
- RemoteModelExperimentalExperimental JAX-backed handle to a served model (policy).
- RemoteVectorEnvExperimentalExperimental JAX-backed remote client for vectorized environments.
- SandboxBuildExperimentalBuild-from-source infrastructure for a sandbox image.
- SandboxEnvExperimentalExperimental JAX-backed owned sandbox session for one environment.
- SandboxInfoExperimentalInformation about a running RLMesh sandbox container.
- SandboxModelExperimentalA model served from an isolated container.
- SandboxRuntimeExperimentalContainer run-time settings for a sandbox --
docker runflags. - SandboxVectorEnvExperimentalExperimental JAX-backed owned sandbox session for vectorized environments.
Functions
- asarray()ExperimentalReturn an immutable JAX array for an RLMesh tensor.
- ensure_available()ExperimentalRaise if JAX is not installed or is older than the supported floor.
- from_array()ExperimentalEncode a JAX array as an RLMesh value.
- space_from_spec()ExperimentalCreate a JAX-adapted space wrapper for a native space spec.