https://github.com/thomashirtz/jax-unet
Minimal implementation of a flexible UNet using JAX and flax.
https://github.com/thomashirtz/jax-unet
Last synced: 4 months ago
JSON representation
Minimal implementation of a flexible UNet using JAX and flax.
- Host: GitHub
- URL: https://github.com/thomashirtz/jax-unet
- Owner: thomashirtz
- Created: 2024-09-10T19:50:29.000Z (about 1 year ago)
- Default Branch: main
- Last Pushed: 2024-11-19T08:14:21.000Z (11 months ago)
- Last Synced: 2025-05-31T07:07:48.834Z (4 months ago)
- Language: Python
- Size: 6.84 KB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# jax-unet
Minimal implementation of a flexible UNet using JAX and flax.