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 fpgamlirfanboy on Sunday April 21, 2024 @02:21PM (#40107946)
> does PyTorch have a similar concept

of course https://github.com/pytorch/pytorch/blob/main/torch/utils/_py...

[ Parent | on HN ]

There are no comments on this comment yet.