3 ms·
I feel like the elephant in the room is that for most cases in this blog post, you either use NumPy arrays or something else like it. You can make ndarray subcl
by toxik 3y ago
I feel like the elephant in the room is that for most cases in this blog post, you either use NumPy arrays or something else like it. You can make ndarray subclasses and that helps but type hinting and NumPy feels so… poorly done.
- patrickkidger 3y agoTry using jaxtyping: https://github.com/google/jaxtyping https://github.com/google/jaxtyping. It also supports numpy/pytorch/etc.