4 projects
numpygrad
A NumPy-only autograd and neural network library with a PyTorch-like API
lipsutils
Internal utilities shared by the Princeton LIPS group.
fibermc
A Jax-based differentiable Monte Carlo estimator with applications to differentiable simulation, computational geometry, and topology optimization.
wavejax
Tools and utilities for computational math and engineering in Jax.