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 _ntka on Sunday April 21, 2024 @02:17PM (#40107907)
Isn't JAX the most widely used framework in the GenAI space? Most companies there use it -- Cohere, Anthropic, CharacterAI, xAI, Midjourney etc.
[ Parent | on HN ]
Re:Penzai: JAX research toolkit for building, editing, and v... by mistrial9 on Sunday April 21, 2024 @05:20PM (#40109343)
just guessing that tech leadership at all of those traces back to Google somehow
[ Parent | on HN ]
Re:Penzai: JAX research toolkit for building, editing, and v... by smhx on Sunday April 21, 2024 @08:03PM (#40110309)
most of the GenAI players use both PyTorch and JAX, depending on the hardware they are running on. Character, Anthro, Midjourney, etc. are dual shops (they use both). xAI only uses JAX afaik.
[ Parent | on HN ]