wassname/jaxtyping
1
0
Fork 0
mirror of https://github.com/wassname/jaxtyping.git synced 2026-08-28 12:50:23 +08:00
Code Issues Packages Projects Releases Wiki Activity
73 Commits 2 Branches 31 Tags
v0.2.14
Commit Graph
14 Commits
Author SHA1 Message Date
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