Re:Simulation of a 2B-atom cell that metabolizes and grows l...
by p1esk on Friday January 28, 2022 @02:29PM
(#30119225)
Why would you suggest JAX? CuPy seems like an obvious choice here (simpler and a lot more mature). Jax is only needed if you want automatic differentiation.
[ Parent
| on HN ]