Patrick Kidger
356f5b7f7b
Build fixes
2023-06-01 11:06:02 -07:00
Patrick Kidger
6a64ef114e
Now provides PyTreeDef, and can detect PyTrees via issubclass(x, PyTree)
2023-06-01 10:56:00 -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
Patrick Kidger
e03c1c329e
We now have Float[np.ndarray, ...] <: np.ndarray. Added basic torch tests. ( #68 )
...
This required quite a lot of refactoring! JAX supports virtual subclass registration (its metaclass is ABCMeta) but NumPy does not, so we have to actually subclass `np.ndarray`.
Simple stuff like __base__ hacking fails due to deallocator conflicts.
2023-03-04 17:29:04 +00:00
Patrick Kidger
81c56052e5
Fixes for some new failures. (Where did they come from?) ( #65 )
...
* Fixes for some new failures. (Where did they come from?)
* Fixed isort?
2023-02-16 10:08:57 -08:00
Patrick Kidger
b5c25822ff
Fixes ( #1 )
...
* Fixes
* Black
* Test fix
* Test fix
* workflow fixes
2022-07-11 13:03:39 +01:00
Patrick Kidger
7ac6ee04a8
initial commit
2022-07-01 21:01:46 +00:00
Patrick Kidger