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


This is a differentiable path tracer written in JAX. Read more about it here: Blog post:

Cornell Box Render

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

You can’t perform that action at this time.