Skip to content
Please note that GitHub no longer supports Internet Explorer.

We recommend upgrading to the latest Microsoft Edge, Google Chrome, or Firefox.

Learn more
Path Tracing in JAX
Jupyter Notebook
Branch: master
Clone or download
Cannot retrieve the latest commit at this time.
Cannot retrieve the latest commit at this time.
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
LICENSE Initial commit Nov 28, 2019
README.md
cornell_box.png initial commit Nov 28, 2019
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.