[NERSC] add jax to desc-python environment

LSSTDESC / desc-help

DESC Computing Requests

BSD 3-Clause "New" or "Revised" License

2 stars 0 forks source link

Closed heather999 closed 1 year ago

heather999 commented 2 years ago

Description A request to add jax to desc-python. See Slack: https://lsstc.slack.com/archives/C2M6H2LAX/p1666124048649539

heather999 commented 2 years ago

First question might be.. just jax or should we also pull in jaxlib?

evevkovacs commented 2 years ago

I only need jax, but is there a downside to including both?

heather999 commented 1 year ago

No downside.. I believe jaxlb is installed when one installs jax anyhow. The latest desc-python includes jax, so I think this can be closed.