jax[cpu]>=0.3.25; platform_system != 'Windows'
torch>=1.1.0; python_version < '3.11'
