NVIDIA / warp

A Python framework for GPU-accelerated simulation, robotics, and machine learning.

pnnl / neuromancer

Pytorch-based framework for solving parametric constrained optimization problems, physics-informed system identification, and parametric model predictive control.

google / jaxopt

Hardware accelerated, batchable and differentiable optimizers in JAX.

optiland / optiland

Comprehensive optical design, optimization, and analysis in Python, including GPU-accelerated and differentiable ray tracing via PyTorch.

deepmodeling / jax-fem

Differentiable Finite Element Method with JAX

metaopt / torchopt

TorchOpt is an efficient library for differentiable optimization built upon PyTorch.