Peter Hawkins 9be96c1d69 Deprecate a number of exports from jax.interpreters.xla. ...

Custom HLO lowering rules for primitives should be updated to use MLIR StableHLO lowering rules via jax.interpreter.mlir.

PiperOrigin-RevId: 561215967

2023-08-29 20:47:59 -07:00

..

__init__.py

Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".

2022-09-22 12:27:19 -07:00

ad.py

Change to simpler import for jax.config

2023-04-21 11:51:22 -07:00

batching.py

Rename Piles to Jumbles, to avoid unfortunate Imperial entanglements.

2023-07-14 15:19:49 -04:00

mlir.py

Make mlir.custom_call() more general and expose it as jax.interpreters.mlir.custom_call().

2023-08-29 08:50:07 -07:00

partial_eval.py

[shape_poly] More cleanup for the internal APIs for shape polymorphism.

2023-07-13 16:37:53 +03:00

pxla.py

Remove unused exports from jax.interpreters.pxla.

2023-08-07 08:27:03 -07:00

xla.py

Deprecate a number of exports from jax.interpreters.xla.

2023-08-29 20:47:59 -07:00