numpy>=1.12
jax
Haiku
