ericjang/pt-jax
Path Tracing in JAX
This project helps researchers and engineers who work with 3D graphics and simulations. It allows you to define complex lighting scenarios and 3D scenes, then efficiently calculate how light interacts within them. The output is physically accurate images and gradients that can be used to optimize scene parameters or material properties.
No commits in the last 6 months.
Use this if you need to simulate realistic light transport in 3D environments and want to use gradients for optimization or inverse problems.
Not ideal if you're looking for a production-ready renderer for real-time applications or purely artistic visualizations without a need for differentiable properties.
Stars
73
Forks
6
Language
Jupyter Notebook
License
MIT
Category
Last pushed
Dec 15, 2019
Commits (30d)
0
Get this data via API
curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/ericjang/pt-jax"
Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.
Higher-rated alternatives
explosion/thinc
🔮 A refreshing functional take on deep learning, compatible with your favorite libraries
google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
patrick-kidger/diffrax
Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable....
google/grain
Library for reading and processing ML training data.
patrick-kidger/equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/