waxmorph

Modules

waxmorph.constants	Numerical constants shared by graph construction and Warp kernels where imported.
waxmorph.data	Normalize meshes and sample their volumes for rollout point clouds.
waxmorph.emulator	Differentiable Warp mechanics and graph diffusion for fixed-size tissue.
waxmorph.gnn	PyTorch graph-network re-exports for the default waxmorph.gnn path.
waxmorph.graph	PyTorch graph re-exports for the default waxmorph.graph path.
waxmorph.jax	Explicit JAX/Equinox backend; top-level waxmorph imports use PyTorch.
waxmorph.losses	PyTorch loss re-exports for the default waxmorph.losses path.
waxmorph.mlp	PyTorch MLP re-exports for the default waxmorph.mlp path.
waxmorph.render	Renderer interfaces for displaying 3D spheroids with morphogens.
waxmorph.simulator	Prescribed Warp mechanics, chemistry, growth, and division in model units.
waxmorph.torch	Default PyTorch backend.

Differentiable morphogenesis and shape assembly with Warp physics.

Top-level learning APIs use waxmorph.torch. Import waxmorph.jax explicitly for the JAX backend.