Setting up an environment with torch, Jax, openmm on CUDA

noegroup / bgflow

Boltzmann Generators and Normalizing Flows in PyTorch

MIT License

150 stars 36 forks source link

Open Olllom opened 2 years ago

Olllom commented 2 years ago

We should provide installation hints to use the jax bridge on CUDA See #40

@jonkhler