News for hackers. Stuff that matters.

A comment on Simulation of a 2B-atom cell that metabolizes and grows like a living cell. Parent comment

Re:Simulation of a 2B-atom cell that metabolizes and grows l... by 4gotunameagain on Friday January 28, 2022 @01:06PM (#30118226)
If you ever need to reap the benefits of CUDA & GPU computations without getting into the details, check out JAX by our corporate overlords™ (https://github.com/google/jax), it has a NumPy like syntax and super fast to get started
[ Parent | on HN ]
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 ]
Re:Simulation of a 2B-atom cell that metabolizes and grows l... by hamilyon2 on Saturday January 29, 2022 @05:52AM (#30125388)
TIL CuPy exists and is stable and mature. This is why i read this forum, every now and then there is serendipitous connection, new angle or discovery.
[ Parent | on HN ]
Re:Simulation of a 2B-atom cell that metabolizes and grows l... by dekhn on Saturday January 29, 2022 @10:31AM (#30127074)
I just want pyXLA, a tool to directly construct XLA programs.
[ Parent | on HN ]