Jake VanderPlas de3191fab3 Cleanup: fix unused imports & mark exported names

2024-10-16 17:42:41 -07:00

..

__init__.py

Cleanup: fix unused imports & mark exported names

2024-10-16 17:42:41 -07:00

BUILD.bazel

Replace pjrt_c_api_gpu_plugin.so symlink with XLA dependency.

2024-08-12 13:01:18 -07:00

plugin_pyproject.toml

[ROCm] Bring-up pjrt support

2024-06-17 16:49:22 +00:00

plugin_setup.py

Update references to the GitHub url in JAX codebase to reflect move from google/jax to jax-ml/jax

2024-09-20 07:52:33 -07:00

pyproject.toml

[ROCm] Bring-up pjrt support

2024-06-17 16:49:22 +00:00

setup.py

Change references in setup.py and utilities to reference the JAX repo move to the JAX-ML org

2024-09-20 07:32:15 -07:00