IMP.jax Namespace Reference

Support for the JAX Python library. More...

Detailed Description

Support for the JAX Python library.

IMP currently has rudimentary support for running on a graphics processing unit (GPU) or similar systems such as Tensor Processing Units (TPUs). This support uses the JAX Python library.

Classes
class	FreeSpace
	An unbounded space with no periodic boundary conditions. More...
class	PeriodicSpace
	A space with periodic boundary conditions. More...
class	Space
	The space in which restraints are evaluated. More...