leap_c.utils.dependencies

Functions

require_jax(→ types.ModuleType)	Import jax lazily and cache it for helpers that require it.
require_torch(→ types.ModuleType)	Import torch lazily and cache it for helpers that require it.

Module Contents

leap_c.utils.dependencies.require_jax() → types.ModuleType

Import jax lazily and cache it for helpers that require it.

leap_c.utils.dependencies.require_torch() → types.ModuleType

Import torch lazily and cache it for helpers that require it.