warp.JaxModulePreloadMode#

class warp.JaxModulePreloadMode(*values)[source]#

Module preload modes for JAX interop callables.

Attributes

NONE = 0#
CURRENT_DEVICE = 1#
ALL_DEVICES = 2#