omnibias.graph.jax.ops
Spectral graph operators and differentiable relaxations for ordering, assignment, and graph-structured learning.
import omnibias.graph.jax.opsModule contract & mathematical conventions
JAX differentiable-graph operator surface.
Shared formulas do not imply universal floating-point identity. Verify dtype, shape, derivative-order support, and the stated numerical assumptions for the path you use.
Imports & aliases
These names share a definition with another public module. Follow a name for its full contract, parameters, and examples.
| Import from this module | Canonical reference |
|---|---|
CutRelaxation | omnibias.graph.jax.ops.CutRelaxation |
degree | omnibias.graph.jax.ops.degree |
fiedler_vector | omnibias.graph.jax.ops.fiedler_vector |
graph_heat_kernel | omnibias.graph.jax.ops.graph_heat_kernel |
graph_laplacian | omnibias.graph.jax.ops.graph_laplacian |
gumbel_sinkhorn | omnibias.graph.jax.ops.gumbel_sinkhorn |
laplacian_spectrum | omnibias.graph.jax.ops.laplacian_spectrum |
normalized_laplacian | omnibias.graph.jax.ops.normalized_laplacian |
random_walk_laplacian | omnibias.graph.jax.ops.random_walk_laplacian |
sample_gumbel | omnibias.graph.jax.ops.sample_gumbel |
sinkhorn_normalize | omnibias.graph.jax.ops.sinkhorn_normalize |
soft_sort | omnibias.graph.jax.ops.soft_sort |
soft_sort_permutation | omnibias.graph.jax.ops.soft_sort_permutation |
soft_top_k | omnibias.graph.jax.ops.soft_top_k |
spectral_clustering_relaxation | omnibias.graph.jax.ops.spectral_clustering_relaxation |
spectral_embedding | omnibias.graph.jax.ops.spectral_embedding |