Skip to content

JAXBackend private

Backend on top of JAX.

Parameters:

Name Type Description Default
enable_x64 bool

Enable 64-bit floating point types.

required

native_t

The native type for tensor data used by the backend.

tensor_t

Types acceptable by the backend API as 'tensors'.

Back to top