jaxtyping

wassname/jaxtyping

Fork 0

mirror of https://github.com/wassname/jaxtyping.git synced 2026-08-30 11:25:15 +08:00

Commit Graph

Select branches

Hide Pull Requests

main

patch-1

v0.0.2

v0.1.0

v0.2.0

v0.2.1

v0.2.10

v0.2.11

v0.2.12

v0.2.13

v0.2.14

v0.2.15

v0.2.16

v0.2.17

v0.2.18

v0.2.19

v0.2.2

v0.2.20

v0.2.21

v0.2.22

v0.2.23

v0.2.24

v0.2.25

v0.2.26

v0.2.27

v0.2.28

v0.2.3

v0.2.4

v0.2.5

v0.2.6

v0.2.7

v0.2.8

v0.2.9

bf241b4e27 We now have e.g. Float[Array, ""] <: Array. Patrick Kidger 2023-03-03 10:32:26 -08:00
5600a1aac8 Fixed cloudpickle breaking, mark 2 Patrick Kidger 2023-03-02 17:37:53 -08:00
2b339715f9 Fixed cloudpickle breaking Patrick Kidger 2023-03-02 12:35:38 -08:00
8c86958b77 Add TypeAlias decoration to PyTree (#66) Zac Cranko 2023-02-28 12:23:47 +11:00
ffc56bf782 Edge case fix Patrick Kidger 2023-02-25 17:38:45 -08:00
5c25da278a Bump version v0.2.13 Patrick Kidger 2023-02-25 17:03:25 -08:00
e2f004afd4 Added support for jax.typing.ArrayLike; now works with PyTorch's bool Patrick Kidger 2023-02-25 17:01:05 -08:00
81c56052e5 Fixes for some new failures. (Where did they come from?) (#65) v0.2.12 Patrick Kidger 2023-02-16 21:08:57 +03:00
d911ebb99c Fix abstractmethods being ignored after @jaxtyped Patrick Kidger 2023-01-22 11:43:36 -08:00
f30b7d1546 Update README.md Patrick Kidger 2023-01-20 07:44:45 -08:00
4b3f834e12 Fix vanilla dataclasses (#56) Brent Yi 2023-01-15 01:52:18 -08:00
59e8fb0d18 Hopefully fixed PyTree raising spurious errors. Bit mysterious that this worked before, really. I've tested this fix as best I can against the various static type checkers, but these are weird and varied enough that this might not be a perfect fix. If you see this and have issues, let me know. (#54) v0.2.11 Patrick Kidger 2022-12-30 19:00:26 +00:00
7dba3516c2 Fixed working with the new (unreleased) version of typeguard (#53) v0.2.10 Patrick Kidger 2022-12-29 17:46:25 +00:00
2b1be5eb0a Update README.md Patrick Kidger 2022-12-07 17:22:57 -08:00
7b3d9a2e9a Explicitly export names in jaxtyping.* (#49) v0.2.9 Brent Yi 2022-12-07 17:21:33 -08:00
29654e7087 JAX is no longer a hard dependency (to support e.g. PyTorch) (#50) Patrick Kidger 2022-12-07 10:48:47 -08:00
8fbf7bf3a5 added link to eqxvision Patrick Kidger 2022-12-05 11:37:32 -08:00
a220df9964 The import hook now decorates dataclass __init__ methods (#48) Patrick Kidger 2022-11-16 13:38:04 -08:00
784aa78f7c update jaxtyped decorator (#44) v0.2.8 Kevin P Murphy 2022-11-14 07:25:28 +01:00
3f877c0dbb Update array_types.py (#41) Patrick Kidger 2022-11-08 22:52:17 -08:00
607f3c66b5 Silenced warning (#40) Patrick Kidger 2022-10-28 15:19:52 -07:00
d3651ca70e NamedTuple example (#36) Peter Roelants 2022-10-03 16:40:51 +02:00
d246e21281 Better import hook (#35) v0.2.7 Patrick Kidger 2022-09-25 23:28:40 -07:00
165065756f Static type-checking fix (#34) v0.2.6 Patrick Kidger 2022-09-24 18:27:22 -07:00
dcd73e3431 Add support for e.g. jaxtyping.Float[Union[...], ...] in py3.8 (#31) ebrevdo 2022-09-22 13:25:29 -07:00
f175c7f315 Fixed py.type not being packaged (#30) v0.2.5 Patrick Kidger 2022-09-22 12:16:25 -07:00
da8300ec6c tidyness tweak (#28) Patrick Kidger 2022-09-20 15:16:52 -07:00
39439c2790 More fixes (#27) v0.2.4 Patrick Kidger 2022-09-20 15:12:21 -07:00
6202dcc639 doc fix (#26) v0.2.3 Patrick Kidger 2022-09-19 23:30:29 -07:00
c2e9d913d5 Fixed jaxtyped breaking descriptors. Fixed long module names. (#25) v0.2.2 Patrick Kidger 2022-09-19 22:40:25 -07:00
62ddcc25b5 Threading fix (#24) Patrick Kidger 2022-09-16 08:04:28 -07:00
a89ebe356b Merge pull request #22 from google/v021 v0.2.1 Patrick Kidger 2022-09-15 08:19:08 -07:00
98133f5e1e Bump version Patrick Kidger 2022-09-15 16:15:28 +01:00
2c7dbbd593 Merge pull request #21 from google/pytyped Patrick Kidger 2022-09-15 08:15:03 -07:00
1291a90192 Merge pull request #20 from google/bool-checking Patrick Kidger 2022-09-15 08:14:56 -07:00
14117804aa Create py.typed Patrick Kidger 2022-09-15 16:07:02 +01:00
e61a37f0a3 Fix for boolean checking not working (#19) Patrick Kidger 2022-09-15 16:05:49 +01:00
3c7e4e11ee Merge pull request #16 from google/rewrite v0.2.0 Patrick Kidger 2022-09-07 12:30:59 -07:00
01f8f20bf5 Fixed a few names Patrick Kidger 2022-09-07 12:20:42 -07:00
cb11a93b22 doc tweaks Patrick Kidger 2022-09-07 12:06:15 -07:00
c82fbbea4e doc tweaks Patrick Kidger 2022-09-07 12:00:58 -07:00
f28b0c789a doc tweaks Patrick Kidger 2022-09-07 11:32:14 -07:00
a53fe6af57 Changes from feedback: Patrick Kidger 2022-09-07 11:09:48 -07:00
140be9ecec Doc updates 2 Patrick Kidger 2022-08-30 13:26:41 -07:00
07e735797e Doc updates Patrick Kidger 2022-08-30 13:14:50 -07:00
48e8131247 Added jaxtyping.Array=jnp.ndarray Patrick Kidger 2022-08-30 12:56:47 -07:00
903000f3d5 Rewrote syntax Patrick Kidger 2022-08-30 12:47:45 -07:00
3f9fad59bd Merge pull request #11 from google/v010 v0.1.0 Patrick Kidger 2022-08-01 01:09:25 +01:00
8f64c99649 Pre-commit fixes Patrick Kidger 2022-08-01 01:06:45 +01:00
3e86c704ae Bump version Patrick Kidger 2022-08-01 01:01:59 +01:00
07259aa5c8 Added support for symbolic expressions in axes. Closes #9. Patrick Kidger 2022-08-01 01:00:18 +01:00
6ff4620d1a Broadcast+variadic is now more lenient, and supports anything broadcastable together. Closes #8 Patrick Kidger 2022-07-31 23:59:20 +01:00
6e2837e5b7 Tidied up _#* into acting as prefixes, regardless of order. Closes #10 Patrick Kidger 2022-07-31 23:37:16 +01:00
7b698ae215 Merge pull request #3 from google/test-tidy Patrick Kidger 2022-07-12 10:11:50 +01:00
e162e1281a Made missing beartype result in test skips rather than tests not running Patrick Kidger 2022-07-12 10:07:42 +01:00
9f75958b2d Make test/helpers.py a relative import Patrick Kidger 2022-07-12 10:03:19 +01:00
81238e38e8 Update tests to make beartype optional Patrick Kidger 2022-07-11 19:27:16 +01:00
35201eb189 Merge pull request #2 from google/fixes2 v0.0.2 Patrick Kidger 2022-07-11 13:15:00 +01:00
0800e5fa38 renamed tests->test so release workflow works Patrick Kidger 2022-07-11 13:06:54 +01:00
b5c25822ff Fixes (#1) Patrick Kidger 2022-07-11 13:03:39 +01:00
0695193b5f workflow fixes Patrick Kidger 2022-07-11 12:58:40 +01:00
5b03d4c29f Test fix Patrick Kidger 2022-07-11 12:55:13 +01:00
47d6b5da9d Test fix Patrick Kidger 2022-07-11 12:53:20 +01:00
f5c10fcbf7 Black Patrick Kidger 2022-07-11 12:50:19 +01:00
60eb17552d Fixes Patrick Kidger 2022-07-11 12:49:57 +01:00
207faaabfd Updated gitignore Patrick Kidger 2022-07-11 12:09:10 +01:00
7ac6ee04a8 initial commit Patrick Kidger 2022-06-24 17:02:18 +00:00

1 2