News for hackers. Stuff that matters.

A comment on Penzai: JAX research toolkit for building, editing, and visualizing neural nets. Parent comment

Re:Penzai: JAX research toolkit for building, editing, and v... by Havoc on Monday April 22, 2024 @03:46AM (#40112182)
Is that gain from TPU usage or something else?
[ Parent | on HN ]
Re:Penzai: JAX research toolkit for building, editing, and v... by MasterScrat on Monday April 22, 2024 @03:49PM (#40118594)
Mostly from the tight JAX-TPU integration yeah
[ Parent | on HN ]