Jake VanderPlas 40367a9eaf Cleanup: remove uses of no-op raise_to_shaped

2024-12-12 09:49:06 -08:00

..

__init__.py

Cleanup: fix unused imports & mark exported names

2024-10-16 17:42:41 -07:00

discharge.py

Merge pull request #25369 from jax-ml:mutable-arrays-ad

2024-12-10 06:36:02 -08:00

indexing.py

[shape_poly] Adding shape polymorphism support for the state primitives.

2024-11-21 06:17:01 -08:00

primitives.py

Add a freeze primitive to delimit ref lifetimes for AD.

2024-12-09 20:57:07 -05:00

types.py

Cleanup: remove uses of no-op raise_to_shaped

2024-12-12 09:49:06 -08:00

utils.py

Cleanup: remove uses of no-op raise_to_shaped

2024-12-12 09:49:06 -08:00