mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-20 22:06:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/lib
History
Peter Hawkins e9c7ff0b7d Deprecate a number of APIs in jax.lib.xla_client. 
(Technically these aren't public, so they don't need a deprecation period, but this is the polite thing to do.)

PiperOrigin-RevId: 684906277
2024-10-11 11:42:40 -07:00
..
__init__.py
Tighten the public API for jax.lib.xla_client & xla_extension
2024-08-03 05:26:22 -07:00
xla_bridge.py
remove jax.extend.backend.default_backend in favor of jax.backend
2024-08-02 07:07:29 -07:00
xla_client.py
Deprecate a number of APIs in jax.lib.xla_client.
2024-10-11 11:42:40 -07:00
xla_extension.py
export jax.lib.xla_extension.HloModule
2024-08-05 14:16:40 -07:00