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

Mono Color

• 1456302503 link to why use type annotations patch-1 Michael J Clark 2024-05-19 09:16:10 +08:00
• 1c0186a1c9 tf, np, torch examples Michael J Clark 2024-05-19 09:12:49 +08:00
• 0d83ee77e6 Updated ecosystem again main Patrick Kidger 2024-04-20 11:25:42 +02:00
• 51ac630ef0 Updated ecosystem Patrick Kidger 2024-04-20 11:18:19 +02:00
• f83170e01d Define the IPython magic lazily Sergei Lebedev 2024-04-16 21:27:01 +01:00
• 07e58de0bd Replace ndim with len(shape) Piotr Kaminski 2024-03-11 15:24:34 +01:00
• 696cc5b90f also require jaxlib Andy Rock 2024-03-07 14:54:20 -05:00
• 102e499d61 Fixes #188. v0.2.28 Patrick Kidger 2024-03-07 17:46:04 +01:00
• f708d1742b Quick fix for docs not generating correctly since the last jaxtyping release v0.2.27 Patrick Kidger 2024-03-02 15:55:14 +01:00
• 5e2518c591 Fix _check_shape str formatting for variadics Alex Ford 2024-03-05 21:23:38 +00:00
• 1b3173ac01 Bump version v0.2.26 Patrick Kidger 2024-02-25 11:55:28 +00:00
• 172b83b4fc Adding a test for generator support (#171) Roman Knyazhitskiy 2024-02-20 20:15:40 +01:00
• 17ea4b13eb No longer imports JAX at all! This is done dynamically when required. See #178 Patrick Kidger 2024-02-18 19:44:22 +00:00
• 9beb5f2d29 Add int4/uint4 support in jaxtyping. (#174) jianlijianli 2024-02-16 17:34:47 -08:00
• 1d4d40294c Added support for beartype 0.17.0's __instancecheck_str__. Patrick Kidger 2024-02-12 01:07:02 +00:00
• 28ad5275d7 Moved print_bindings into storage.py Patrick Kidger 2024-02-12 00:03:51 +00:00
• d7fd59a34c Added print_bindings. Patrick Kidger 2024-01-08 14:17:32 -08:00
• 8de8c0bb68 Correct pytree path in array.md Afroz Mohiuddin 2024-02-11 18:31:20 -08:00
• f18de2ce28 Added better docs on stringified type annotations Patrick Kidger 2024-01-08 04:31:27 -08:00
• eb9a23df63 Update dataclass docs (#155) Patrick Kidger 2024-01-05 05:17:45 -08:00
• adf1a5e4e3 chore(docs): fix typos in docstrings Jérome Eertmans 2024-01-04 11:13:15 +01:00
• 272be74e01 Bump version v0.2.25 Patrick Kidger 2023-12-15 18:26:45 +00:00
• 7df267efa4 Upgrade to ruff-format Patrick Kidger 2023-12-10 15:27:42 -08:00
• d43933f942 Updated to latest pyktdocs_tweaks Patrick Kidger 2023-12-09 14:55:30 -08:00
• 1acc0d7153 Improved error messages a little bit, in particular to highlight individual problematic arguments. Patrick Kidger 2023-12-05 19:32:41 -08:00
• 33cf4fcdac Simplified internals by removing jaxtyping_raise; jaxtyping_malformed. Patrick Kidger 2023-12-02 17:44:57 -08:00
• e5cc75e4a3 Removed internal jaxtyped_fns registry that is no longer needed. Patrick Kidger 2023-12-02 17:30:30 -08:00
• 125bc89ee9 Added environment config flags. Patrick Kidger 2023-12-02 17:26:55 -08:00
• 8e47c9081c version bump v0.2.24 Patrick Kidger 2023-11-27 09:46:52 -08:00
• 205978958f Fix install_import_hook(..., None) Patrick Kidger 2023-11-06 10:33:59 -08:00
• 850f4e72cd Handle beartype doing isinstance(None, hint) Patrick Kidger 2023-11-02 20:01:54 -07:00
• 0a76c9c70c Error message improvements Patrick Kidger 2023-11-02 20:01:27 -07:00
• 5fbd6718ab Added support for 'self' in dataclass attribute annotations; switched from args and kwargs to just arguments. Patrick Kidger 2023-11-02 20:01:05 -07:00
• 80a99568f7 Removed unused elements from unions, e.g. Float[ArrayLike, ...] will no longer include Float[np.bool, ...]. Patrick Kidger 2023-11-02 19:59:33 -07:00
• baffbef5ca Doc fix Patrick Kidger 2023-11-02 19:54:38 -07:00
• 127eae56b7 array shapes+dtypes no longer checked as part of pytree flattening. This avoids edge-case crash when using pytree-path dependent sizes Patrick Kidger 2023-11-02 19:56:46 -07:00
• 58600d3fe0 Fixed new-style PRNG keys. Patrick Kidger 2023-10-28 13:51:15 -07:00
• 7925e278f4 Symbolic expressions now support delayed binding to arguments. Fixes #93. Patrick Kidger 2023-10-22 21:27:16 -07:00
• ba3b2027cc Standardised terminology: now using just "axis"/"axes", not "dimension" Patrick Kidger 2023-10-23 09:58:16 -07:00
• 12d540794f Pretty error messages: fixes #6. Patrick Kidger 2023-10-22 20:35:38 -07:00
• 63e0fdff74 Typecheck errors now state the size of the stored axis and structure values. Patrick Kidger 2023-10-22 10:07:43 -07:00
• d12291de7e Added support for treepath-dependent sizes. Patrick Kidger 2023-10-21 19:53:54 -07:00
• 9e1ba8a77d Added support for declaring PyTree structures. Patrick Kidger 2023-10-21 15:24:20 -07:00
• e4a93ee218 Added with jaxtyped("context"):, for now undocumented. Patrick Kidger 2023-10-24 10:25:42 -07:00
• 260fb36876 Fixed mixing variadic+broadcast with variadic+nonbroadcast dimensions. Patrick Kidger 2023-10-21 12:56:16 -07:00
• 9646eff7e1 Fixes for g3 Patrick Kidger 2023-10-23 10:05:31 -07:00
• 1a048b1f2f Fixed Float[ArrayLike, "#*foo"] leaving out bool/int/float/complex. Patrick Kidger 2023-10-21 11:49:28 -07:00
• 338ca631c6 beartype error messages for dataclass attributes now use the correct name Patrick Kidger 2023-10-18 17:10:49 -07:00
• 2ab8286c81 Error message for bad symbolic shapes is now useful. Patrick Kidger 2023-10-18 16:58:56 -07:00
• 7f85a12a85 Added DTypeLike to match jax.typing Patrick Kidger 2023-10-15 13:16:00 -07:00
• e55348a4b4 Added jaxtyping.Real Patrick Kidger 2023-10-15 12:06:28 -07:00
• 7a84b27da9 Bump version v0.2.23 Patrick Kidger 2023-10-10 21:40:38 +01:00
• 77c263c3de Allow only typeguard lower than 3.x.x version Roma Knyaz 2023-10-10 17:50:51 +02:00
• 91a36aaee4 dataclasses now have fields checked, not __init__. Patrick Kidger 2023-09-27 16:05:37 -07:00
• 513a54b048 Better handling of user-defined typechecker Roma Knyaz 2023-10-07 23:21:05 +02:00
• 9c9635d4f3 Add Orbax to ecosystem list. Patrick Kidger 2023-10-06 15:51:22 +01:00
• c3e7fd35a2 Update ecosystem links Patrick Kidger 2023-10-06 15:43:29 +01:00
• ef102f40b4 Bump up pre-commit ruff and black versions Roma Knyaz 2023-10-02 21:22:00 +02:00
• 4917c2e30f Fix flakiness of transitive import hook test Roma Knyaz 2023-10-02 17:17:32 +02:00
• 17092ad8d8 Simplified the import hook tests Patrick Kidger 2023-09-27 11:12:30 -07:00
• 75392d6330 Added missing license headers Patrick Kidger 2023-09-27 09:11:57 -07:00
• 18b8e76d67 Be tolerant of faulty IPython installs. Patrick Kidger 2023-09-26 02:11:14 +01:00
• 1e5229c20e Should be more robust to jax/numpy/tensorflow version changes Patrick Kidger 2023-09-22 10:58:55 -07:00
• e05985df2b Document IPython extension and version nump v0.2.22 Patrick Kidger 2023-09-20 11:12:10 -07:00
• f454cb797c Make jaxtyping an IPython extension Roma Knyaz 2023-09-20 11:02:39 +02:00
• d2785baced Better suppotr for incomplete JAX installations. Supersedes #105. Patrick Kidger 2023-09-14 20:09:23 -07:00
• c80c1264d3 Error message now mentions both array type and shape requirements. Supersedes #107. Patrick Kidger 2023-09-14 20:05:45 -07:00
• 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 Patrick Kidger 2023-09-14 16:29:24 -07:00
• e347c480d5 Hash is now stable across python runtimes v0.2.21 Patrick Kidger 2023-08-17 08:15:32 -07:00
• 13e6870fb8 Compatibility with JAX changes to opaque dtypes Patrick Kidger 2023-07-26 09:30:40 -07:00
• 4c90808401 jaxtyping no longer appears in tracebacks Patrick Kidger 2023-07-26 09:26:59 -07:00
• 5a57456e15 document Levanter Patrick Kidger 2023-07-12 11:31:46 -07:00
• a6ab6c0d28 Have PRNGKeyArray appear correctly in downstream documentation Patrick Kidger 2023-06-29 18:39:22 -07:00
• 83be9e9d16 Fixed jaxtyping doc generation Patrick Kidger 2023-06-25 12:00:16 -07:00
• d2aa9c1e8d Merge branch 'main' of https://github.com/google/jaxtyping Patrick Kidger 2023-06-14 10:42:32 -07:00
• 926dc53856 Have PyTreeDef appear correctly in docs Patrick Kidger 2023-06-14 10:42:14 -07:00
• edc34f14f8 Update ecosystem links. Patrick Kidger 2023-06-07 15:35:45 +01:00
• 8fa15050bc Update ecosystem links. Patrick Kidger 2023-06-07 15:35:14 +01:00
• 356f5b7f7b Build fixes v0.2.20 Patrick Kidger 2023-06-01 11:06:02 -07:00
• 1b9c9fab52 Bump to Py3.9 Patrick Kidger 2023-06-01 10:47:33 -07:00
• 066a5b058f Made modules private. Patrick Kidger 2023-05-30 12:34:10 -07:00
• 319d54abcf Avoid __builtins__ getting added as a key Patrick Kidger 2023-05-30 12:30:07 -07:00
• 6a64ef114e Now provides PyTreeDef, and can detect PyTrees via issubclass(x, PyTree) Patrick Kidger 2023-05-15 12:16:53 -07:00
• 10e1852b37 Fix favicon Patrick Kidger 2023-05-12 11:57:06 -07:00
• a19149d23d Fixed pytest hook v0.2.19 Patrick Kidger 2023-05-11 09:08:09 -07:00
• 0c596ff373 Fix for non-JAX installations. v0.2.18 Patrick Kidger 2023-05-10 17:16:42 -07:00
• 7934d2afed Static typing fixes v0.2.17 Patrick Kidger 2023-05-10 16:34:27 -07:00
• 849b15db3b Added better docs; added PRNGKeys. Packaging with pyproject.toml. v0.2.16 Patrick Kidger 2023-05-10 13:44:22 -07:00
• 46c7896c99 Transitive test now uses typeguard instead of beartype Patrick Kidger 2023-04-19 11:48:08 -07:00
• c92b0d0ab1 Some improvements (#77) v0.2.15 Patrick Kidger 2023-04-13 11:18:55 -07:00
• 158b8b8f0c Now works with torch.compile? (#72) Patrick Kidger 2023-04-13 10:53:14 -07:00
• 9b6df18b83 Update FAQ to mention ruff Patrick Kidger 2023-03-19 22:21:21 +00:00
• f0b240df5f Switched to ruff Patrick Kidger 2023-03-15 22:36:24 -07:00
• a4d27c7cc1 Fixed _Jaxtyped.__get__, e.g. swallowing abstractmethod decorations Patrick Kidger 2023-03-15 22:27:36 -07:00
• ee46c57e53 Version bump v0.2.14 Patrick Kidger 2023-03-05 20:26:01 -08:00
• 38be24f9c8 beartype+inheritance fix. Bool[int, '...'] now correctly raises an error. Patrick Kidger 2023-03-05 20:09:49 -08:00
• e718f00cc5 Fixed import hook hitting __pycache__ even when you change the choice of runtime type checker Patrick Kidger 2023-03-05 16:19:44 -08:00
• c232eeaa89 Fixed pytest plugin with new import hook typechecker syntax Patrick Kidger 2023-03-05 12:19:14 -08:00
• e03c1c329e We now have Float[np.ndarray, ...] <: np.ndarray. Added basic torch tests. (#68) Patrick Kidger 2023-03-04 17:29:04 +00:00
• fef81cf0a0 The import hook now supports BeartypeConf/BeartypeStrategy Patrick Kidger 2023-03-03 10:34:03 -08:00