API REFERENCE/omnibias-binary

omnibias.binary.jax.ops

0.1.0a2AlphaPython >=3.10Apache-2.0

Binary and multilevel quantization with explicit hard-forward and surrogate-gradient contracts.

import omnibias.binary.jax.ops

Module contract & mathematical conventions

JAX quantization-gradient 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.

Module source ↗Source revision 9a2d73df0573Running reference examples →

quantizeModule ↗surrogateModule ↗

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
binarize	omnibias.binary.jax.ops.binarize
binarize01	omnibias.binary.jax.ops.binarize01
binarize_curvature	omnibias.binary.jax.ops.binarize_curvature
curvature_corrected_slope	omnibias.binary.jax.ops.curvature_corrected_slope
heaviside	omnibias.binary.jax.ops.heaviside
kbit_quantize	omnibias.binary.jax.ops.kbit_quantize
riccati_sigmoid_derivative	omnibias.binary.jax.ops.riccati_sigmoid_derivative
riccati_tanh_derivative	omnibias.binary.jax.ops.riccati_tanh_derivative
surrogate_jet	omnibias.binary.jax.ops.surrogate_jet
surrogate_tower	omnibias.binary.jax.ops.surrogate_tower
ternarize	omnibias.binary.jax.ops.ternarize

← Package overviewRunnable learning examples →