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'.