Implicit Neural Representations
WIP. Implicit Neural Representations in JAX with DMHaiku.
Requirements
pip install --upgrade jax jaxlib pillow optax
# for a GPU enviroment,
# pip install -U jax jaxlib==0.1.62+cuda110 -f https://storage.googleapis.com/jax-releases/jax_releases.html
pip install git+https://github.com/deepmind/dm-haiku
COIN
python coin.py