Update pyproject.toml to include platform dependent requirements

[jgostick]

Apparently it's possible to specify in the pyproject.toml what versions of certain requirements to install on each platform, and perhaps even to exclude some packages. For instance:

"jax>=0.4.18,<0.5.0; sys_platform == 'darwin'",

There is some discussion here