mirror of
https://github.com/ROCm/jax.git
synced 2025-04-18 12:56:07 +00:00
f939ac078d
Updates XLA to 00afc7bb81.
The new XLA release removes the use of protocol buffers from the XLA client. Fixes #349.
Add backward compatibility shims to jaxlib to allow older jax releases to still work on an up to date jaxlib.
The new XLA release also incorporates a fix that avoids a host-device copy for every iteration of a `lax.fori_loop()` on GPU. Fixes #402.
Add a new jaxlib.__version__ field, change jax/jaxlib compatibility logic to check for it.
12 lines
107 B
Plaintext
12 lines
107 B
Plaintext
*.pyc
|
|
*.so
|
|
*.egg-info
|
|
.ipynb_checkpoints
|
|
/bazel-*
|
|
.bazelrc
|
|
/tensorflow
|
|
.DS_Store
|
|
build/
|
|
dist/
|
|
.mypy_cache/
|