You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
This project uses Patrick Kidger's torch typing to give tensor types including shapes. He recommends using JAXTyping for newer projects (not JAX specific, better maintained, more compatible with type checkers, etc) And it'd be good to update TL to use it.
@dkamm might be up your alley? Sorry if this invalidates your enum work, but might be a more elegant solution!
The text was updated successfully, but these errors were encountered:
This project uses Patrick Kidger's torch typing to give tensor types including shapes. He recommends using JAXTyping for newer projects (not JAX specific, better maintained, more compatible with type checkers, etc) And it'd be good to update TL to use it.
@dkamm might be up your alley? Sorry if this invalidates your enum work, but might be a more elegant solution!
The text was updated successfully, but these errors were encountered: