Jake VanderPlas a009e1cf50 deprecate jax.lib.xla_client.bfloat16

2024-08-06 11:22:27 -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 jax.lib.xla_client.bfloat16

2024-08-06 11:22:27 -07:00

xla_extension.py

export jax.lib.xla_extension.HloModule

2024-08-05 14:16:40 -07:00