Getting Started
Reference Documentation
Advanced JAX Tutorials
Notes
custom_vjp
nondiff_argnums
Developer documentation
API documentation
Warning
This page was created from a pull request (#9655).
Returns the platform name of the default XLA backend.
str
previous
jax.lib package
next
jax.lib.xla_bridge.device_count