jax
tfp-nightly

[tests]
flake8
ruff
pytest
scipy
