7 Commits
Author SHA1 Message Date
Patrick Kidger e2f004afd4 Added support for jax.typing.ArrayLike; now works with PyTorch's bool 2023-02-25 17:01:27 -08:00
Patrick Kidger 29654e7087 JAX is no longer a hard dependency (to support e.g. PyTorch) (#50) 2022-12-07 10:48:47 -08:00
Kevin P Murphy 784aa78f7c update jaxtyped decorator (#44)
* update jaxtyped decorator

* add newline character to pacify flake

* add precomit hooks

* add missing return statement

* replace typing_extensions>=4.2.0 with typing_extensions

* pin version range for typing_extensions

* set min version of typing-extensions but not max

* bump version number to 0.2.8
2022-11-13 22:25:28 -08:00
Patrick Kidger f175c7f315 Fixed py.type not being packaged (#30) 2022-09-22 12:16:25 -07:00
Patrick Kidger 6ff4620d1a Broadcast+variadic is now more lenient, and supports anything broadcastable together. Closes #8 2022-07-31 23:59:20 +01: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