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...

Functions
def	get_random_key
	Get a new JAX random key seeded from IMP's random number generator. More...

Function Documentation

def IMP.jax.get_random_key

(

)

Get a new JAX random key seeded from IMP's random number generator.

Note

This function is only available in Python.

Definition at line 74 of file jax.py.