Dan Foreman-Mackey
dda6430f7c
Add register_custom_call_target to xla_client API docs.
...
This function is (for better or worse) user facing for custom call
users. I think it's worth having this in the API docs.
2024-06-27 14:40:36 -04:00
Peter Hawkins
74f1ab0503
Export Device as jax.Device.
...
Users are writing things like jax.lib.xla_client.Device in type annotations which is not a public API. Add a supported public name for the Device type.
2023-02-02 12:58:15 -05:00
Jake VanderPlas
179f5ab200
Fix documentation broken by 8dc1dff6
2023-01-31 15:23:31 -08:00
Jake VanderPlas
81e627d5bd
DOC: make API doc titles more uniform
2023-01-18 10:59:42 -08:00
Jake VanderPlas
91bad73cac
DOC: remove reference to TpuDevice
2022-12-13 11:00:10 -08:00
Sharad Vikram
1daea700f2
Bump JAX/Jaxlib versions
2022-06-28 14:36:47 -07:00
Peter Hawkins
1af5c888bd
Remove CpuDevice from documentation index.
...
The corresponding class no longer exists.
2022-04-16 10:03:47 -04:00
Peter Hawkins
95f47074da
Remove xla_bridge.{constant, register_constant_handler, _python_scalar_constant} from API.
...
An upcoming change will move and rename these functions, and it's not clear they should have been public in the first place.
PiperOrigin-RevId: 404051961
2021-10-18 13:56:58 -07:00
Jake VanderPlas
ff2bfc0e87
DOC: fix docstring and add docs.
2021-09-27 09:48:27 -07:00
