Matthew Johnson 3f9eb404e4 remove named_shapes (since xmap is now gone)

2024-07-25 00:54:50 +00:00

..

__init__.py

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

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

functions.py

Fix error message in jax.nn.dot_product_attention when the inputs have different dtypes.

2024-07-24 07:13:15 -07:00

initializers.py

remove named_shapes (since xmap is now gone)

2024-07-25 00:54:50 +00:00