Differentiable Python Packages

mellon

Non-parametric density inference for single-cell analysis.

datarax

A Differentiable Data Pipeline Framework for JAX

torch-contour

Differentiable contour to image operations with PyTorch

diffbio

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

differt2d

2D Toolbox for Differentiable Ray Tracing

croissant-sim

CROISSANT: A differentiable visibility simulator based on spherical harmonics

pastax

**P**arameterizable **A**uto-differentiable **S**imulators of ocean **T**rajectories in j**AX**.

gradoptics

Differentiable Optics via Ray Tracing

jax-dips

JAX-DIPS is a differentiable interfacial PDE solver.