wassname/jaxtyping
1
0
Fork 0
mirror of https://github.com/wassname/jaxtyping.git synced 2026-08-27 12:10:13 +08:00
Code Issues Packages Projects Releases Wiki Activity
167 Commits 2 Branches 31 Tags
patch-1
Commit Graph
2 Commits
Author SHA1 Message Date
Patrick Kidger e03c1c329e We now have Float[np.ndarray, ...] <: np.ndarray. Added basic torch tests. (#68) 
This required quite a lot of refactoring! JAX supports virtual subclass registration (its metaclass is ABCMeta) but NumPy does not, so we have to actually subclass `np.ndarray`.
Simple stuff like __base__ hacking fails due to deallocator conflicts.
 2023-03-04 17:29:04 +00:00
Patrick Kidger 7ac6ee04a8 initial commit 2022-07-01 21:01:46 +00:00