Differentiable Python Packages

mellon

Non-parametric density inference for single-cell analysis.

datarax

Differentiable data pipelines for JAX/Flax NNX: HuggingFace, TFDS and ArrayRecord sources, augmentation stages, scan-based epochs, exact mid-epoch resume.

diffbio

End-to-end differentiable bioinformatics for JAX/Flax NNX: soft Smith-Waterman alignment, read mapping and assembly, variant calling, RNA-seq, single-cell, epigenomics, CRISPR, metabolomics, multi-omics, protein and RNA structure, molecular dynamics and drug-discovery operators composed into trainable pipelines on the Avitai stack.

differt2d

2D Toolbox for Differentiable Ray Tracing

torch-contour

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

croissant-sim

A differentiable visibility simulator based on spherical harmonics

jax-dips

JAX-DIPS is a differentiable interfacial PDE solver.

gradoptics

Differentiable Optics via Ray Tracing