JAX AND Sparse models
Common descendants