-r requirements_common.txt

# Dependencies only required for Ubuntu
# Needs to be addressed. Unblocked 2.4 branchcut cl/338377048
PyYAML ~= 6.0

# Test dependencies which don't exist on Windows
jax ~= 0.3.14
jaxlib ~= 0.4.1; platform.machine != 'aarch64'
