jactorch.functional

Functional APIs for JacTorch. By default, you should just access functions with jactorch.*. Or, if you want to be explicit about the functions, you can import the specific function from jactorch.functional.*.

Submodules

jactorch.functional.arith	Arithmetic operations.
jactorch.functional.clustering	Clustering functions.
jactorch.functional.grad	Gradient modification functions.
jactorch.functional.indexing	Tensor indexing utils.
jactorch.functional.kernel	Useful utilities for kernel-based attention mechanism.
jactorch.functional.linalg	Linear algebra functions.
jactorch.functional.loglinear	Linear algebra functions in the log space.
jactorch.functional.masking	Masking-related functions.
jactorch.functional.probability	Probability distributions related functions.
jactorch.functional.quantization	Basic quantization functions with a straight-through gradient estimator.
jactorch.functional.range	Numeric range functions.
jactorch.functional.sampling	Sampling functions.
jactorch.functional.shape	Tensor shape utilities.