Updated 6 months ago
pycharge
Differentiable electromagnetics simulation library for moving point charges built on JAX
Updated 6 months ago
https://github.com/ceyron/exponax
Efficient Differentiable n-d PDE solvers in JAX.
Updated 5 months ago
https://github.com/ctu-vras/monoforce
[IROS 2024] [ICML 2024 Workshop Differentiable Almost Everything] MonoForce: Learnable Image-conditioned Physics Engine
Updated 6 months ago
https://github.com/ceyron/trainax
Training methodologies for autoregressive neural operators in JAX.