ericjang / pt-jax
Path Tracing in JAX
☆69Updated 4 years ago
Related projects: ⓘ
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 4 years ago
- Supplementary materials for our SIGGRAPH 2022 paper☆27Updated 2 years ago
- ☆97Updated 2 months ago
- Pytrees + dataclasses ❤️☆60Updated 2 months ago
- ☆12Updated 3 years ago
- Dive into Jax, Flax, XLA and C++☆31Updated 4 years ago
- ☆78Updated 3 years ago
- Texture mapping with variational auto-encoders☆40Updated 2 years ago
- Automatic Differentiation for high-performance stencil loops☆12Updated 3 years ago
- We simulate a wind tunnel, place a rectangular occlusion in it, and then use gradient descent to turn the occlusion into a wing.☆25Updated 3 years ago
- Normalizing Flows in Jax☆105Updated 4 years ago
- A sparse KLU solver for PyTorch.☆60Updated 2 years ago
- Prototypes of differentiable differential equation solvers in JAX.☆27Updated 4 years ago
- Hookean springs in PyTorch☆50Updated last year
- A differentiation API for PyTorch☆29Updated 4 years ago
- ☆24Updated 5 years ago
- A raytracer written in PyTorch (raynet?)☆240Updated 4 years ago
- Geometric Algebra package for JAX☆47Updated 2 years ago
- Jax-based MaxEnt☆18Updated 4 years ago
- ☆22Updated 4 years ago
- Very deep VAEs in JAX/Flax☆45Updated 3 years ago
- FluidNet re-written with ATen tensor lib☆51Updated 5 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆34Updated 3 years ago
- Riemannian Convex Potential Maps☆68Updated last year
- A differentiable programming language with an integration primitive that soundly handles interactions among the derivative, integral, and…☆36Updated 5 months ago
- Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrende…☆58Updated 10 months ago
- Convex potential flows☆78Updated 2 years ago
- Documentation:☆119Updated last year
- Code for "Efficient optimization of loops and limits with randomized telescoping sums"☆27Updated 5 years ago