Skip to content
Path Tracing in JAX
Jupyter Notebook
Branch: master
Clone or download
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
LICENSE Initial commit Nov 28, 2019
README.md initial commit Nov 28, 2019
cornell_box.png
cornell_box_pt_numpy.ipynb initial commit Nov 28, 2019
jaxpt.ipynb
jaxpt_vmap.ipynb updated code to use pmap on multi-device TPU Nov 29, 2019

README.md

pt-jax

This is a differentiable path tracer written in JAX. Read more about it here: Blog post: https://blog.evjang.com/2019/11/jaxpt.html

Cornell Box Render

Dependencies: JAX, Matplotlib, Jupyter Notebook or Google Colaboratory Notebook

You can’t perform that action at this time.