Skip to content

Add JaxModule: enables using jax functions in PyTorch autograd graphs #1

Add JaxModule: enables using jax functions in PyTorch autograd graphs

Add JaxModule: enables using jax functions in PyTorch autograd graphs #1

Annotations

1 warning

The logs for this run have expired and are no longer available.