jax
flax

[dev]
black
pre-commit
jaxgl
pygame
