15 Commits
Author SHA1 Message Date
Patrick Kidger c92b0d0ab1 Some improvements (#77)
* Various improvements.

- Added support for functions in symbolic dimensions, e.g. "min(foo,bar)", which were previously disallowed due to the presence of a comma. (#51)
- Added support for adding ignored names to dimensions, e.g. "cols=4". (#76)

* Now works with Python 3.10 A | B union types.
2023-04-13 19:18:55 +01:00
Patrick Kidger fef81cf0a0 The import hook now supports BeartypeConf/BeartypeStrategy 2023-03-03 10:34:03 -08:00
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 a220df9964 The import hook now decorates dataclass __init__ methods (#48) 2022-11-16 13:38:04 -08:00
Patrick Kidger cb11a93b22 doc tweaks 2022-09-07 12:06:15 -07:00
Patrick Kidger c82fbbea4e doc tweaks 2022-09-07 12:00:58 -07:00
Patrick Kidger f28b0c789a doc tweaks 2022-09-07 11:32:14 -07:00
Patrick Kidger a53fe6af57 Changes from feedback:
- Float32 introduced to replace f32 (etc.)
- Old f32 aliases were previously around for backward-compatibility, but
  the code is pretty hideous, and all stakeholders are now on board
  through RFC #13.
- Overall feedback was that Int{Sign,Unsign} was too long and not enough
  like numpy. Changed to Int, UInt, and Integer.
2022-09-07 11:09:48 -07:00
Patrick Kidger 140be9ecec Doc updates 2 2022-08-30 13:26:41 -07:00
Patrick Kidger 07e735797e Doc updates 2022-08-30 13:14:50 -07:00
Patrick Kidger 48e8131247 Added jaxtyping.Array=jnp.ndarray 2022-08-30 12:56:47 -07:00
Patrick Kidger 903000f3d5 Rewrote syntax 2022-08-30 12:47:45 -07:00
Patrick Kidger 07259aa5c8 Added support for symbolic expressions in axes. Closes #9. 2022-08-01 01:00:18 +01:00
Patrick Kidger 6e2837e5b7 Tidied up _#* into acting as prefixes, regardless of order. Closes #10 2022-07-31 23:45:40 +01:00
Patrick Kidger 7ac6ee04a8 initial commit 2022-07-01 21:01:46 +00:00