Dan Foreman-Mackey 96045043a4 Move ir_attribute builder from extend.ffi to interpreters.mlir. ...

While this function is currently only used for lowering FFI calls, it could be used most places where `ir.*Attr` objects are directly constructed.

PiperOrigin-RevId: 661761712

2024-08-11 01:47:49 -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

Finalize deprecations of jax.interpreters.ad config & source_info_util

2024-03-18 13:33:17 -07:00

batching.py

del add_any_p and zeros_like_p, replace aval-dispatched traceable

2023-12-21 17:04:21 -08:00

mlir.py

Move ir_attribute builder from extend.ffi to interpreters.mlir.

2024-08-11 01:47:49 -07:00

partial_eval.py

integrate attrs in jax.jit

2024-01-27 17:44:43 -08:00

pxla.py

Make pxla.shard_arg batch calls to xc.copy_array_to_devices_with_sharding

2024-06-13 13:10:10 -07:00

xla.py

Deprecate jax.interpreters xb, xc, xe abbreviations.

2024-06-28 12:38:43 -07:00