mononet

mononet — Unconstrained monotonic neural networks.

Multi-backend support for PyTorch, JAX (Flax NNX), and Keras 3. See https://arxiv.org/abs/2205.11775 for the reference paper.

Backends are imported lazily: import mononet does not import torch / jax / keras. Use from mononet.torch import ... (or the equivalent for jax/keras) to access backend layers.

Subpackages

• mononet.core
  • mononet.core.config
  • mononet.core.reference
  • mononet.core.init
  • mononet.core.numerics
  • mononet.core.types
• mononet.torch
  • mononet.torch.layers
• mononet.jax
  • mononet.jax.layers
• mononet.keras
  • mononet.keras.layers