Peter Hawkins 6fc78a5a6d Deprecate jax.lax.infeed and jax.lax.outfeed. ...

These APIs are already broken on GPU and TPU by virtue of not being implemented in the PJRT C API, so it seems unlikely that they have any users.

PiperOrigin-RevId: 746595857

2025-04-11 14:42:14 -07:00

..

__init__.py

Deprecate jax.lax.infeed and jax.lax.outfeed.

2025-04-11 14:42:14 -07:00

linalg.py

Start adding primitive registration helper functions to lax.linalg.

2025-02-21 04:05:34 -08:00