dml-rs
Fast double / debiased machine learning for causal inference, implemented in Rust.
dml-rs implements the double / debiased machine learning framework of Chernozhukov et al. (2018) with a scikit-learn style API. Every model is tested for numerical agreement with the reference DoubleML package, and the Rust core makes estimation typically 3–24× faster.
Installation
pip install dml-rs
Prebuilt wheels are available for Linux, macOS and Windows (Python 3.9 or newer).
Quick start
import numpy as np
import dml_rs
# Simulated data: the true effect of d on y is 0.5; X confounds both
rng = np.random.default_rng(42)
n = 2000
X = rng.normal(size=(n, 5))
d = np.sin(X[:, 0]) + X[:, 1] ** 2 + rng.normal(size=n)
y = 0.5 * d + np.cos(X[:, 0]) + X[:, 1] ** 2 + rng.normal(size=n)
model = dml_rs.PLR(
ml_l=dml_rs.RandomForestRegressor(min_samples_leaf=5), # learner for E[y | X]
ml_m=dml_rs.RandomForestRegressor(min_samples_leaf=5), # learner for E[d | X]
n_folds=5,
)
model.fit(X, y, d)
print(model.summary)
coef std err t P>|t| 2.5 % 97.5 %
d 0.516919 0.029474 17.538215 7.319031e-69 0.459151 0.574687
Any scikit-learn estimator works as a learner. The built-in learners
(GradientBoostingRegressor, RandomForestRegressor, Lasso, LassoCV, Ols,
LogisticRegression, LogisticRegressionCV, and the corresponding classifiers) run the whole estimation in Rust, in parallel and without the GIL,
and are scikit-learn estimators themselves (GridSearchCV, Pipeline, clone and tuning work with them).
Models
| Class | Target parameter |
|---|---|
PLR |
Partially linear regression: effect of a continuous or binary treatment |
IRM |
Interactive regression model: ATE / ATTE of a binary treatment; cate and gate |
PLIV |
Partially linear IV regression |
IIVM |
Interactive IV model: LATE with a binary instrument |
DID, DIDCS |
Difference-in-differences for panels and repeated cross-sections |
StaggeredDID |
Staggered adoption: ATT(g, t) with group, time and event-study aggregation |
SLearner, TLearner, XLearner, RLearner, DRLearner |
Heterogeneous (conditional average) treatment effects |
All estimators support repeated sample splitting, multiplier bootstrap with simultaneous
confidence bands (bootstrap, confint(joint=True)), multiple-testing adjustment (p_adjust),
pandas summaries, and the scikit-learn parameter protocol (get_params, set_params, clone).
PLR, IRM and DID models also offer sensitivity analysis for omitted confounders
(sensitivity_analysis, sensitivity_summary, sensitivity_benchmark). Pass cluster= to
fit for cluster-robust inference with one- or two-way clustered data, and inspect the nuisance
learners with nuisance_loss_ and evaluate_learners.
Documentation
Full feature overview, benchmarks and validation details: https://github.com/AsadGudzevic94/dml-rs
License
Licensed under either of the Apache License 2.0 or the MIT license, at your option.
1 maintainer
Metadata
1 maintainer
Release files for dml-rs 1.4.0
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | Provenance | Actions |
|---|---|---|---|---|
| dml_rs-1.4.0.tar.gz | 512.7 kB | With provenance | Details for dml_rs-1.4.0.tar.gz |
Built distributions (wheels)
| File | Actions Reset | |||
|---|---|---|---|---|
| dml_rs-1.4.0-cp39-abi3-win_amd64.whl | CPython 3.9 | abi3 | Windows x86-64 | Details for dml_rs-1.4.0-cp39-abi3-win_amd64.whl |
| dml_rs-1.4.0-cp39-abi3-manylinux2014_x86_64.manylinux_2_17_x86_64.whl | CPython 3.9 | abi3 | Linux glibc 2.17+ x86-64 | Details for dml_rs-1.4.0-cp39-abi3-manylinux2014_x86_64.manylinux_2_17_x86_64.whl |
| dml_rs-1.4.0-cp39-abi3-manylinux2014_aarch64.manylinux_2_17_aarch64.whl | CPython 3.9 | abi3 | Linux glibc 2.17+ ARM64 | Details for dml_rs-1.4.0-cp39-abi3-manylinux2014_aarch64.manylinux_2_17_aarch64.whl |
| dml_rs-1.4.0-cp39-abi3-macosx_11_0_arm64.whl | CPython 3.9 | abi3 | macOS 11.0+ ARM64 | Details for dml_rs-1.4.0-cp39-abi3-macosx_11_0_arm64.whl |
| dml_rs-1.4.0-cp39-abi3-macosx_10_12_x86_64.whl | CPython 3.9 | abi3 | macOS 10.12+ x86-64 | Details for dml_rs-1.4.0-cp39-abi3-macosx_10_12_x86_64.whl |
Total release size: 12.1 MB
Release files / dml_rs-1.4.0.tar.gz
| Download URL | dml_rs-1.4.0.tar.gz |
|---|---|
| Size | 512.7 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
d93e20ef88c144fb7e4c1067c16eb6567f2149065755305defd1fefa32d7ee65
|
|
BLAKE2b-256 checksum How to use checksums |
d1d276765dd000027a586de6583bb2a94f92a9524e5c6a8a428bc86e815d1f84
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.16
|
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 Nov 4, 2026.
Transparency logRelease files / dml_rs-1.4.0-cp39-abi3-win_amd64.whl
| Download URL | dml_rs-1.4.0-cp39-abi3-win_amd64.whl |
|---|---|
| Size | 2.3 MB |
| Tags | CPython 3.9 Windows x86-64 abi3 |
|
SHA-256 checksum How to use checksums |
2b9d3f5214ead5d7363df35f3c46e993548d84f7949e90515aa2aec9580883a2
|
|
BLAKE2b-256 checksum How to use checksums |
37f585df5e0700ba67433af9a0a0d6b5e3a3ca032a982fcd89d8827091182f43
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.16
|
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 Nov 4, 2026.
Transparency logRelease files / dml_rs-1.4.0-cp39-abi3-manylinux2014_x86_64.manylinux_2_17_x86_64.whl
| Download URL | dml_rs-1.4.0-cp39-abi3-manylinux2014_x86_64.manylinux_2_17_x86_64.whl |
|---|---|
| Size | 2.4 MB |
| Tags | CPython 3.9 Linux glibc 2.17+ x86-64 abi3 |
|
SHA-256 checksum How to use checksums |
8c0cfb8802aa0e54345a3bf360e31b2992009107796201c979618a3967543691
|
|
BLAKE2b-256 checksum How to use checksums |
56f23da1e6f8ddd52552a48d6f210be2eb80f88075df4c45d4c807ec2cce4dc8
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.16
|
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 Nov 4, 2026.
Transparency logRelease files / dml_rs-1.4.0-cp39-abi3-manylinux2014_aarch64.manylinux_2_17_aarch64.whl
| Download URL | dml_rs-1.4.0-cp39-abi3-manylinux2014_aarch64.manylinux_2_17_aarch64.whl |
|---|---|
| Size | 2.4 MB |
| Tags | CPython 3.9 Linux glibc 2.17+ ARM64 abi3 |
|
SHA-256 checksum How to use checksums |
62c48ef2e0917560292409f34512bef7c34514b391a9c0bd25e75e4d34fd457d
|
|
BLAKE2b-256 checksum How to use checksums |
5e1062507b4f0b5be689369f299caf1ea22394cd80c3191c7a06015e407264b1
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.16
|
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 Nov 4, 2026.
Transparency logRelease files / dml_rs-1.4.0-cp39-abi3-macosx_11_0_arm64.whl
| Download URL | dml_rs-1.4.0-cp39-abi3-macosx_11_0_arm64.whl |
|---|---|
| Size | 2.2 MB |
| Tags | CPython 3.9 abi3 macOS 11.0+ ARM64 |
|
SHA-256 checksum How to use checksums |
1e6b4ce516cf44db997e529b22f1359b7d5c73b1178f5e14864caeff6fccdffc
|
|
BLAKE2b-256 checksum How to use checksums |
9874b77f3e28d2276c38030c902aceee92449dae525358ebd424f079e9c2bdec
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.16
|
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 Nov 4, 2026.
Transparency logRelease files / dml_rs-1.4.0-cp39-abi3-macosx_10_12_x86_64.whl
| Download URL | dml_rs-1.4.0-cp39-abi3-macosx_10_12_x86_64.whl |
|---|---|
| Size | 2.3 MB |
| Tags | CPython 3.9 abi3 macOS 10.12+ x86-64 |
|
SHA-256 checksum How to use checksums |
a70fbe605118e51264425e0211b76206be893ca638f02a5f5c5ffd40ff43c133
|
|
BLAKE2b-256 checksum How to use checksums |
973794a4c8778d089322ccdde00701f87c2ef11801febf158c7b37a2029f0d59
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
Yes |
| Uploaded via |
twine/7.0.0 CPython/3.13.16
|
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 Nov 4, 2026.
Transparency logSecurity for dml-rs 1.4.0
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.
Full provenance
All 6 files in this release have been uploaded with provenance metadata.
Attestation for dml_rs-1.4.0.tar.gz
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 Nov 4, 2026.
Transparency logA note on package security
PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.
To learn more about protecting yourself from malicious actors, visit our package security documentation.
