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 MasterScrat on Sunday April 21, 2024 @02:38PM (#40108084)
We've built our startup from scratch on JAX, selling text-to-image model finetuning, and it's given us a consistent edge not only in terms of pure performance but also in terms of "dollars per unit of work"
[ Parent | on HN ]
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 ]