Grigory SapunovJAXJAX by Google Research is getting more and more popular. Deepmind recently announced they are using JAX to accelerate their research and…Dec 20, 20201Dec 20, 20201