e3nn-jax
latest
API
Tutorial
Benchmarks
e3nn-jax
Index
Edit on GitHub
Index
A
|
B
|
C
|
D
|
E
|
F
|
G
|
I
|
L
|
M
|
N
|
P
|
Q
|
R
|
S
|
T
|
U
|
V
|
W
|
X
|
Z
A
angles_to_axis_angle() (in module e3nn_jax)
angles_to_matrix() (in module e3nn_jax)
angles_to_quaternion() (in module e3nn_jax)
angles_to_xyz() (in module e3nn_jax)
apply() (e3nn_jax.SphericalSignal method)
as_irreps_array() (in module e3nn_jax)
assert_equivariant() (in module e3nn_jax.utils)
assert_output_dtype_matches_input_dtype() (in module e3nn_jax.utils)
astype() (e3nn_jax.IrrepsArray method)
axis_angle_to_angles() (in module e3nn_jax)
axis_angle_to_matrix() (in module e3nn_jax)
axis_angle_to_quaternion() (in module e3nn_jax)
axis_to_irreps() (e3nn_jax.IrrepsArray method)
axis_to_mul() (e3nn_jax.IrrepsArray method)
B
BatchNorm (class in e3nn_jax.flax)
(class in e3nn_jax.haiku)
bessel() (in module e3nn_jax)
broadcast_to() (e3nn_jax.IrrepsArray method)
C
chunks (e3nn_jax.IrrepsArray property)
clebsch_gordan() (in module e3nn_jax)
compose_angles() (in module e3nn_jax)
compose_axis_angle() (in module e3nn_jax)
compose_quaternion() (in module e3nn_jax)
concatenate() (in module e3nn_jax)
count() (e3nn_jax.Irreps method)
cross() (in module e3nn_jax)
D
D_from_angles() (e3nn_jax.Irrep method)
(e3nn_jax.Irreps method)
D_from_log_coordinates() (e3nn_jax.Irrep method)
(e3nn_jax.Irreps method)
D_from_matrix() (e3nn_jax.Irrep method)
(e3nn_jax.Irreps method)
D_from_quaternion() (e3nn_jax.Irrep method)
(e3nn_jax.Irreps method)
dim (e3nn_jax.Irrep property)
(e3nn_jax.Irreps attribute)
(e3nn_jax.Irreps property)
dot() (in module e3nn_jax)
Dropout (class in e3nn_jax.haiku)
dtype (e3nn_jax.IrrepsArray property)
(e3nn_jax.SphericalSignal property)
E
elementwise_tensor_product() (in module e3nn_jax)
equivariance_test() (in module e3nn_jax.utils)
extend_with_zeros() (e3nn_jax.IrrepsArray method)
F
filter() (e3nn_jax.Irreps method)
(e3nn_jax.IrrepsArray method)
find_peaks() (e3nn_jax.SphericalSignal method)
from_chunks() (in module e3nn_jax)
from_function() (e3nn_jax.SphericalSignal static method)
from_s2grid() (in module e3nn_jax)
FunctionalTensorProduct (class in e3nn_jax.legacy)
G
gate() (in module e3nn_jax)
generators() (e3nn_jax.Irrep method)
(e3nn_jax.Irreps method)
(in module e3nn_jax)
grad() (in module e3nn_jax)
grid_alpha (e3nn_jax.SphericalSignal property)
grid_resolution (e3nn_jax.SphericalSignal property)
grid_vectors (e3nn_jax.SphericalSignal property)
grid_y (e3nn_jax.SphericalSignal property)
I
identity_angles() (in module e3nn_jax)
identity_quaternion() (in module e3nn_jax)
index() (e3nn_jax.Irreps method)
integrate() (e3nn_jax.SphericalSignal method)
inverse_angles() (in module e3nn_jax)
inverse_quaternion() (in module e3nn_jax)
Irrep (class in e3nn_jax)
Irreps (class in e3nn_jax)
IrrepsArray (class in e3nn_jax)
is_scalar() (e3nn_jax.Irrep method)
(e3nn_jax.Irreps method)
iterator() (e3nn_jax.Irrep class method)
L
left_right() (e3nn_jax.legacy.FunctionalTensorProduct method)
Linear (class in e3nn_jax.flax)
(class in e3nn_jax.haiku)
lmax (e3nn_jax.Irreps attribute)
(e3nn_jax.Irreps property)
ls (e3nn_jax.Irreps attribute)
(e3nn_jax.Irreps property)
M
matrix_to_angles() (in module e3nn_jax)
matrix_to_axis_angle() (in module e3nn_jax)
matrix_to_quaternion() (in module e3nn_jax)
matrix_x() (in module e3nn_jax)
matrix_y() (in module e3nn_jax)
matrix_z() (in module e3nn_jax)
mean() (in module e3nn_jax)
mul_gcd (e3nn_jax.Irreps property)
mul_to_axis() (e3nn_jax.IrrepsArray method)
MultiLayerPerceptron (class in e3nn_jax.flax)
(class in e3nn_jax.haiku)
N
ndim (e3nn_jax.IrrepsArray property)
(e3nn_jax.SphericalSignal property)
norm() (in module e3nn_jax)
norm_activation() (in module e3nn_jax)
normal() (in module e3nn_jax)
normalize_function() (in module e3nn_jax)
num_irreps (e3nn_jax.Irreps attribute)
(e3nn_jax.Irreps property)
P
pad_to_plot() (e3nn_jax.SphericalSignal method)
plotly_surface() (e3nn_jax.SphericalSignal method)
poly_envelope() (in module e3nn_jax)
Q
quadrature_weights (e3nn_jax.SphericalSignal property)
quaternion_to_angles() (in module e3nn_jax)
quaternion_to_axis_angle() (in module e3nn_jax)
quaternion_to_matrix() (in module e3nn_jax)
R
radius_graph() (in module e3nn_jax)
rand_angles() (in module e3nn_jax)
rand_axis_angle() (in module e3nn_jax)
rand_matrix() (in module e3nn_jax)
rand_quaternion() (in module e3nn_jax)
rechunk() (e3nn_jax.IrrepsArray method)
reduced_antisymmetric_tensor_product_basis() (in module e3nn_jax)
reduced_symmetric_tensor_product_basis() (in module e3nn_jax)
reduced_tensor_product_basis() (in module e3nn_jax)
regroup() (e3nn_jax.Irreps method)
(e3nn_jax.IrrepsArray method)
remove_zero_chunks() (e3nn_jax.IrrepsArray method)
remove_zero_multiplicities() (e3nn_jax.Irreps method)
repeat() (e3nn_jax.Irreps method)
repeat_irreps_by_last_axis() (e3nn_jax.IrrepsArray method)
replace_values() (e3nn_jax.SphericalSignal method)
res_alpha (e3nn_jax.SphericalSignal property)
res_beta (e3nn_jax.SphericalSignal property)
resample() (e3nn_jax.SphericalSignal method)
reshape() (e3nn_jax.IrrepsArray method)
right() (e3nn_jax.legacy.FunctionalTensorProduct method)
S
s2_dirac() (in module e3nn_jax)
s2_irreps() (in module e3nn_jax)
sample() (e3nn_jax.SphericalSignal method)
scalar_activation() (in module e3nn_jax)
scatter_max() (in module e3nn_jax)
scatter_mean() (in module e3nn_jax)
scatter_sum() (in module e3nn_jax)
set_mul() (e3nn_jax.Irreps method)
sh() (in module e3nn_jax)
shape (e3nn_jax.IrrepsArray property)
(e3nn_jax.SphericalSignal property)
simplify() (e3nn_jax.Irreps method)
(e3nn_jax.IrrepsArray method)
slice_by_chunk (e3nn_jax.Irreps property)
(e3nn_jax.IrrepsArray property)
slice_by_dim (e3nn_jax.Irreps property)
(e3nn_jax.IrrepsArray property)
slice_by_mul (e3nn_jax.Irreps property)
(e3nn_jax.IrrepsArray property)
slices() (e3nn_jax.Irreps method)
soft_envelope() (in module e3nn_jax)
soft_one_hot_linspace() (in module e3nn_jax)
sort() (e3nn_jax.Irreps method)
(e3nn_jax.IrrepsArray method)
spherical_harmonics() (e3nn_jax.Irreps static method)
(in module e3nn_jax)
SphericalSignal (class in e3nn_jax)
sum() (in module e3nn_jax)
sus() (in module e3nn_jax)
SymmetricTensorProduct (class in e3nn_jax.haiku)
T
tensor_product() (in module e3nn_jax)
tensor_product_with_spherical_harmonics() (in module e3nn_jax)
tensor_square() (in module e3nn_jax)
to_s2grid() (in module e3nn_jax)
to_s2point() (in module e3nn_jax)
transform_by_angles() (e3nn_jax.IrrepsArray method)
(e3nn_jax.SphericalSignal method)
transform_by_axis_angle() (e3nn_jax.IrrepsArray method)
(e3nn_jax.SphericalSignal method)
transform_by_log_coordinates() (e3nn_jax.IrrepsArray method)
transform_by_matrix() (e3nn_jax.IrrepsArray method)
(e3nn_jax.SphericalSignal method)
transform_by_quaternion() (e3nn_jax.IrrepsArray method)
(e3nn_jax.SphericalSignal method)
U
unify() (e3nn_jax.Irreps method)
(e3nn_jax.IrrepsArray method)
V
vmap() (in module e3nn_jax.utils)
W
where() (in module e3nn_jax)
X
xyz_to_angles() (in module e3nn_jax)
Z
zeros() (e3nn_jax.SphericalSignal static method)
(in module e3nn_jax)
zeros_like() (in module e3nn_jax)
Read the Docs
v: latest
Versions
latest
stable
Downloads
On Read the Docs
Project Home
Builds