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 eli_gottlieb on Sunday April 21, 2024 @03:05PM (#40108264)
If JAX had affine_grid() and grid_sample(), I'd be using it instead of PyTorch for my current project.
[ Parent | on HN ]

There are no comments on this comment yet.