e3x.nn.wrappers

Convenience wrappers to simplify usage of other functions.

Functions

basis(r, *, max_degree, num, radial_fn[, ...])

Convenience wrapper for computing radial-angular basis functions.

Classes

AngularFn(*args, **kwargs)	Protocol for angular functions.
ExponentialBasis([initial_gamma, ...])	Exponential basis module.