3 ms·
Thanks for the updated script! I added a comment to my code pointing to yours. I’m curious what the key slowdown in my attempt was.
by brrrrrm 5y ago
Thanks for the updated script! I added a comment to my code pointing to yours. I’m curious what the key slowdown in my attempt was.
- patrickkidger 5y agoNot at all -- I was very happy to see some Equinox code out in the wild! I think the main slowdown was doing the model updates out of the JIT region. JAX-without-JIT is ridiculously slow.