mononet

Unconstrained monotonic neural networks with first-class support for PyTorch, JAX (Flax NNX), and Keras 3.

Reference implementation of:

Runje, D., Shankaranarayana, S. M. (2023). Constrained Monotonic Neural Networks. ICML 2023. arXiv:2205.11775

Install

pip install "mononet[torch]"      # PyTorch
pip install "mononet[jax]"        # JAX + Flax NNX
pip install "mononet[keras]"      # Keras 3
pip install "mononet[all]"        # all three

Citation

If you use mononet in academic work, please cite the reference paper:

@inproceedings{runje2023constrained,
  title     = {Constrained Monotonic Neural Networks},
  author    = {Runje, Davor and Shankaranarayana, Sharath M.},
  booktitle = {Proceedings of the 40th International Conference on Machine Learning},
  series    = {Proceedings of Machine Learning Research},
  volume    = {202},
  year      = {2023},
  publisher = {PMLR},
  url       = {https://proceedings.mlr.press/v202/runje23a.html},
  eprint    = {2205.11775},
  archivePrefix = {arXiv}
}

Note: confirm the exact BibTeX entry against the PMLR proceedings page before the first PyPI release — venue, volume, and URL fields are sensitive to typos.