ericjang / pt-jaxLinks
Path Tracing in JAX
☆71Updated 5 years ago
Alternatives and similar repositories for pt-jax
Users that are interested in pt-jax are comparing it to the libraries listed below
Sorting:
- Supplementary materials for our SIGGRAPH 2022 paper☆28Updated 3 years ago
- A raytracer written in PyTorch (raynet?)☆244Updated 5 years ago
- Experiment code for "Randomized Automatic Differentiation"☆67Updated 5 years ago
- Hookean springs in PyTorch☆54Updated 9 months ago
- We simulate a wind tunnel, place a rectangular occlusion in it, and then use gradient descent to turn the occlusion into a wing.☆26Updated 4 years ago
- ☆105Updated last year
- Automatic Differentiation for high-performance stencil loops☆12Updated 4 years ago
- Documentation:☆121Updated 2 years ago
- Pytrees + dataclasses ❤️☆67Updated 3 months ago
- FluidNet re-written with ATen tensor lib☆52Updated 6 years ago
- Prototypes of differentiable differential equation solvers in JAX.☆27Updated 5 years ago
- Normalizing Flows in Jax☆107Updated 4 years ago
- Texture mapping with variational auto-encoders☆40Updated 3 years ago
- A differentiation API for PyTorch☆30Updated 5 years ago
- A differentiable LCP physics engine in PyTorch.☆305Updated 4 years ago
- ☆12Updated 4 years ago
- Interpolation routines in Pytorch.☆110Updated 4 years ago
- ☆214Updated 3 years ago
- B-Spline CNNs on Lie groups☆54Updated 5 years ago
- A differentiable programming language with an integration primitive that soundly handles interactions among the derivative, integral, and…☆37Updated last year
- Numerical integration methods for mass-springs systems using PyTorch's autodiff☆31Updated 9 months ago
- Fusing Taichi into PyTorch☆145Updated 2 years ago
- A sparse KLU solver for PyTorch.☆65Updated 3 years ago
- ☆68Updated last year
- Understanding ML and deep learning through geometry☆157Updated 3 years ago
- Neural reparameterization improves structural optimization☆123Updated 3 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆112Updated 3 years ago
- Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.☆56Updated 2 weeks ago
- Very deep VAEs in JAX/Flax☆46Updated 4 years ago
- ☆80Updated 3 years ago