binned-cdf
A PyTorch-based distribution parametrized by the logits of CDF bins
Background
The Cumulative Distribution Function (CDF) is a fundamental concept in probability theory and statistics that describes the probability that a random variable $X$ takes on a value less than or equal to a given threshold $x$. Formally, the CDF is defined as $F(x) = P(X \leq x)$, where $F(x)$ ranges from 0 to 1 as $x$ varies from negative to positive infinity. The CDF provides a complete characterization of the probability distribution of a random variable: for continuous distributions, it is the integral of the probability density function (PDF), while for discrete distributions, it is the sum of probabilities up to and including $x$. Key properties of any CDF are the monotonicity and the boundary conditions $\lim_{x \to -\infty} F(x) = 0$ and $\lim_{x \to \infty} F(x) = 1$. CDFs are particularly useful for computing probabilities of intervals, quantiles, and for statistical inference.
Application to Machine Learning
This repository uses the CDF to model and learn flexible probability distributions in machine learning tasks. By parameterizing the CDF with binned logits, it enables differentiable training and efficient sampling, making it suitable for uncertainty estimation, probabilistic prediction, and distributional modeling in neural networks.
Implementation
The PiecewiseConstantBinnedCDF and PiecewiseLinearBinnedCDF classes inherit directly from
torch.distributions.Distribution, implementing all necessary methods plus some convenience functions.
They support multi-dimensional batch shapes and CUDA devices.
The bins can be initialized linearly or log-spaced.
torch>=2.7 it the only non-dev dependency of this repo.
Getting Started
I recommend using PiecewiseLinearBinnedCDF for most applications.
from binned_cdf import PiecewiseLinearBinnedCDF
distr = PiecewiseLinearBinnedCDF(
logits=logits, # shape: (*batch_shape, num_bins)
bound_low=-5, # adapt to your data
bound_up=7, # adapt to your data
log_spacing=True, # if False, linear spacing is used
normalization_method="sigmoid", # "sigmoid" or "softmax"
)
# ... use it like any other torch.distribution.Distribution
👉 Please have a look at the documentation to get started.
1 maintainer
Metadata
1 maintainer
Release files for binned-cdf 2.1.4
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| binned_cdf-2.1.4.tar.gz | 41.6 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| binned_cdf-2.1.4-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 59.7 kB
Release files / binned_cdf-2.1.4.tar.gz
| Download URL | binned_cdf-2.1.4.tar.gz |
|---|---|
| Size | 41.6 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
a494e2990af3f9ec8821f4c2c7341a87b0b213a23fe5645442d5cf412fab07c1
|
|
BLAKE2b-256 checksum How to use checksums |
f456869ac2f8e013b9a080113af52cc2fa5c14324b98383afcc6ff71e08178d7
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Oct 16, 2026.
Transparency logRelease files / binned_cdf-2.1.4-py3-none-any.whl
| Download URL | binned_cdf-2.1.4-py3-none-any.whl |
|---|---|
| Size | 18.1 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
2c943524db3278747184e16f8521b257de2470123a80c512d918c162a05dde6a
|
|
BLAKE2b-256 checksum How to use checksums |
e32ddc510f399702b92e8b33f8dcb1cf0c764df063b87128dec186f1a6d6a379
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.14
|
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
PyPI Publish Attestation
PyPI verified that this artifact, at this checksum, originated from the publisher listed below.
Signed by GitHub Actions, verified by PyPI on Oct 16, 2026.
Transparency log
