GitHub repository: nmboffi/jax-edm2
Primary language: Python
Description: An implementation of NVIDIA's EDM2 network in jax.