What are you using and why?
Great article. Thank you for sharing. I knew academia was more into pytorch, but this gives some real proof.
Throwing Jax in at the end there really through me for a loop! I hadn’t even thought of higher-order gradient schemes or pushing numpy maths onto the GPU. Will definitely be using a few functions from that library.