mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 21:06:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jax/_src/lax
History
Roy Frostig 1c84e4a753 migrate internal dependencies from jax.interpreters.batching to jax._src.interpreters.batching 
... in preparation for paring down `jax.interpreters.batching`'s exported symbols.

PiperOrigin-RevId: 508487887
2023-02-09 15:11:57 -08:00
..
control_flow
simpler pretty-print for pjit, tweak custom pp rule signature
2023-02-09 12:45:51 -08:00
__init__.py
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
ann.py
migrate internal dependencies from jax.interpreters.batching to jax._src.interpreters.batching
2023-02-09 15:11:57 -08:00
convolution.py
migrate internal dependencies from jax.interpreters.batching to jax._src.interpreters.batching
2023-02-09 15:11:57 -08:00
eigh.py
[TPU] Add cutoff for nearly diagonal matrices in QDWH-eigh algorithm.
2022-11-23 08:17:50 -08:00
fft.py
migrate internal dependencies from jax.interpreters.batching to jax._src.interpreters.batching
2023-02-09 15:11:57 -08:00
lax.py
migrate internal dependencies from jax.interpreters.batching to jax._src.interpreters.batching
2023-02-09 15:11:57 -08:00
linalg.py
migrate internal dependencies from jax.interpreters.batching to jax._src.interpreters.batching
2023-02-09 15:11:57 -08:00
other.py
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
parallel.py
migrate internal dependencies from jax.interpreters.batching to jax._src.interpreters.batching
2023-02-09 15:11:57 -08:00
qdwh.py
migrate internal dependencies from jax.core to jax._src.core
2022-12-16 21:00:14 -08:00
slicing.py
migrate internal dependencies from jax.interpreters.batching to jax._src.interpreters.batching
2023-02-09 15:11:57 -08:00
stack.py
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
2022-09-22 12:27:19 -07:00
svd.py
migrate internal dependencies from jax.core to jax._src.core
2022-12-16 21:00:14 -08:00
utils.py
migrate internal dependencies from jax.core to jax._src.core
2022-12-16 21:00:14 -08:00
windowed_reductions.py
migrate internal dependencies from jax.interpreters.batching to jax._src.interpreters.batching
2023-02-09 15:11:57 -08:00