Skip to content

v0.1.0

Choose a tag to compare

@JimGalasyn JimGalasyn released this 15 Jul 19:17

First release of jax-morpho: a differentiable, GPU-scale developmental morphogenesis engine in JAX.

  • Center-based tissue engine (Morse potential, autodiff relaxation, division) reproducing the Gibson-2006 epithelium polygon-side distribution.
  • O(N) jax_md neighbor-list scaling — 1–2M cells on a single GPU.
  • Gradient-based inverse design of tissue shape, and a genome→form map differentiable to the genome.

MIT licensed. See README for install and demos.