2 Commits

Include renames

Author	SHA1	Message	Date
Patrick Kidger	e308695293	Updated to support both new and old style JAX PRNG keys, as they are going to co-exist simultaneously. See https://github.com/google/jax/pull/17297	2023-09-14 20:00:46 -07:00
Patrick Kidger	849b15db3b	Added better docs; added PRNGKeys. Packaging with pyproject.toml. ... Also added support for matching dtypes using regexes, and for nesting e.g. Shaped[Float[Array, "dim1 dim2"], "dim3"].	2023-05-10 15:27:25 -07:00