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

[PJRT C API] Rename the folder "plugins" to "jax_plugins".

2023-11-14 13:56:13 -08:00

plugin_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

pyproject.toml

[PJRT C API] Rename the folder "plugins" to "jax_plugins".

2023-11-14 13:56:13 -08: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