One-command support for JAX GPU

lnccbrown / HSSM

Development of HSSM package

Other

70 stars 10 forks source link

Closed digicosmos86 closed 1 month ago

digicosmos86 commented 1 month ago

After this pr is merged, one can install hssm with GPU support with pip install hssm[cuda12] (not available on cuda)

AlexanderFengler commented 1 month ago

not available on conda :)

digicosmos86 commented 1 month ago

Correct. conda lol