UNet-in-JAX. Simple 1d UNet in JAX & Equinox to solve the Poisson equation.

github.com/Ceyron/UNet-in-JAX

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.