Differentiable Python Packages

mellon

Non-parametric density inference for single-cell analysis.

datarax

A Differentiable Data Pipeline Framework for JAX

croissant-sim

A differentiable visibility simulator based on spherical harmonics

torch-contour

python torch library for polygon / contour to image (mask, distance map) processing and more

differt2d

2D Toolbox for Differentiable Ray Tracing

gradoptics

Differentiable Optics via Ray Tracing

pastax

Parameterized Auto-differentiable Simulator for ocean Trajectories in jAX.

jax-dips

JAX-DIPS is a differentiable interfacial PDE solver.

diffbio

End-to-end differentiable bioinformatics pipelines built on JAX/Flax NNX