tensorflow
jax[cpu]
namex
black>=22
flake8
isort
pytest