Sheaf – A Functional Language for Differentiable Programs (sheaf-lang.org)

I've been building a functional language for differentiable programming that compiles to JAX. The core idea is homoiconicity applied to ML, models are data structures that can inspect and transform themselves.