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

PiperOrigin-RevId: 676843138

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

..

__init__.py

Reapply: move jax.interpreters.ad to jax._src.interpreters.ad

2023-02-02 09:29:05 -08:00

ad.py

Clean up and fix primal type to tangent type mapping

2024-09-18 13:43:54 -07:00

batching.py

Reverts 66a3f87a24016594794c2ee289826baed5e979a4

2024-08-20 15:13:09 -07:00

mlir.py

Add frontend attributes to Jax. This allows Jax users to annotate Jax code with frontend_attributes which can be traced down to the HLO level, to be used for numerical debugging purposes.

2024-09-06 16:44:56 -07:00

partial_eval.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

pxla.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

xla.py

Run pyupgrade --py310-plus.

2024-06-26 16:10:18 -04:00