IMP.algebra.jax Namespace Reference

Utility functions for the JAX Python library. More...

Detailed Description

Utility functions for the JAX Python library.

This provides various algebraic and geometric functions that are similar to those in IMP.algebra but are implemented using the JAX Python library. They are intended to support code in other IMP modules that is implemented in JAX.

Functions
def	get_random_vector_in_3d_sphere
	Return a random vector uniformly distributed in a 3-dimensional sphere of given radius at the origin. More...
def	get_random_vector_in_sphere
	Return a random vector uniformly distributed in an n-dimensional sphere of given radius at the origin. More...
def	get_random_vector_on_3d_sphere
	Return a random vector uniformly distributed on the surface of a 3-dimensional sphere of given radius at the origin. More...
def	get_rotation_about_normalized_axis
	Return a quaternion from the given normalized axis and angle. More...

Function Documentation

def IMP.algebra.jax.get_random_vector_in_3d_sphere	(		key,
			radius,
			shape = 1
	)

Return a random vector uniformly distributed in a 3-dimensional sphere of given radius at the origin.

Note

This function is only available in Python.

Definition at line 17 of file algebra/jax.py.

def IMP.algebra.jax.get_random_vector_in_sphere	(		key,
			n,
			radius
	)

Return a random vector uniformly distributed in an n-dimensional sphere of given radius at the origin.

Note

This function is only available in Python.

Definition at line 35 of file algebra/jax.py.

def IMP.algebra.jax.get_random_vector_on_3d_sphere	(		key,
			radius,
			shape = 1
	)

Return a random vector uniformly distributed on the surface of a 3-dimensional sphere of given radius at the origin.

Note

This function is only available in Python.

Definition at line 60 of file algebra/jax.py.

def IMP.algebra.jax.get_rotation_about_normalized_axis	(		axis_norm,
			angle
	)

Return a quaternion from the given normalized axis and angle.

Note

This function is only available in Python.

Definition at line 76 of file algebra/jax.py.