Search Packages

differt

Differentiable Ray Tracing Toolbox for Radio Propagation Simulations

cheetah-accelerator

Fast and differentiable particle accelerator optics simulation for reinforcement learning and optimisation applications.

phiflow

A differentiable PDE solving framework for machine learning

pyrddlgym-jax

pyRDDLGym-jax: automatic differentiation for solving sequential planning problems in JAX.

agent-torch

large population models

jwave

A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs

jaxsso

A framework for structural shape optimization based on automatic differentiation (AD) and the adjoint method, enabled by JAX

socialforce

PyTorch implementation of DeepSocialForce.

softtorch

A library for soft differentiable relaxations of common PyTorch functions.