Skip to main content

dml-rs (Python)

PyPI Python versions CI

Fast double / debiased machine learning for causal inference, implemented in Rust and validated against DoubleML.

pip install dml-rs
import dml_rs
from sklearn.ensemble import RandomForestRegressor, RandomForestClassifier

# Partially linear regression with any scikit-learn learners
plr = dml_rs.PLR(ml_l=RandomForestRegressor(), ml_m=RandomForestRegressor(), n_folds=5, n_rep=3)
plr.fit(X, y, d)
print(plr.summary)

# Built-in Rust learners run the whole cross-fitting in parallel without the GIL
plr = dml_rs.PLR(ml_l=dml_rs.Lasso(alpha=0.01), ml_m=dml_rs.Lasso(alpha=0.01)).fit(X, y, d)

# Average treatment effect of a binary treatment
irm = dml_rs.IRM(ml_g=RandomForestRegressor(), ml_m=RandomForestClassifier(), score="ATE").fit(X, y, d)

# Joint inference over several treatments
plr.bootstrap(method="normal", n_boot=1000)
plr.confint(joint=True)
plr.p_adjust("romano-wolf")

See the repository for details.

Metadata

Release files for dml-rs 0.5.0

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for dml-rs 0.5.0
File Size Uploaded Provenance
dml_rs-0.5.0.tar.gz 101.3 kB With provenance Details

Built distributions (wheels)

Table of built distributions (wheels) for dml-rs 0.5.0
File
dml_rs-0.5.0-cp39-abi3-win_amd64.whl CPython 3.9 abi3 Windows x86-64 Details
dml_rs-0.5.0-cp39-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl CPython 3.9 abi3 Linux glibc 2.17+ x86-64 Details
dml_rs-0.5.0-cp39-abi3-manylinux_2_17_aarch64.manylinux2014_aarch64.whl CPython 3.9 abi3 Linux glibc 2.17+ ARM64 Details
dml_rs-0.5.0-cp39-abi3-macosx_11_0_arm64.whl CPython 3.9 abi3 macOS 11.0+ ARM64 Details
dml_rs-0.5.0-cp39-abi3-macosx_10_12_x86_64.whl CPython 3.9 abi3 macOS 10.12+ x86-64 Details

Total release size: 3.6 MB

Release files / dml_rs-0.5.0.tar.gz

Download URL dml_rs-0.5.0.tar.gz
Size 101.3 kB
Tags Source
SHA-256 checksum
How to use checksums
62c36eeb9f35461a18fc389166133722c2290011ca238b2114bd53f27183c326
BLAKE2b-256 checksum
How to use checksums
0201041e16573466eb4c2e8555c9ce5e3674f340a3111dcec220dc154ecc5e90
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 Oct 23, 2026.

Transparency log

Release files / dml_rs-0.5.0-cp39-abi3-win_amd64.whl

Download URL dml_rs-0.5.0-cp39-abi3-win_amd64.whl
Size 591.8 kB
Tags CPython 3.9 Windows x86-64 abi3
SHA-256 checksum
How to use checksums
a31d0a0e30728558dde2b3ce3a03c9b1643d021c255c982d735944f13db9d30d
BLAKE2b-256 checksum
How to use checksums
208a17f33b0a731b4da2ab45395f6197b8b6f580b94d3590e72cd9122d817485
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 Oct 23, 2026.

Transparency log

Release files / dml_rs-0.5.0-cp39-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

Download URL dml_rs-0.5.0-cp39-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Size 774.2 kB
Tags CPython 3.9 Linux glibc 2.17+ x86-64 abi3
SHA-256 checksum
How to use checksums
3d1af583db19c7665b708454775c275ef7263f79b2645e8f977e1f3da8a5a196
BLAKE2b-256 checksum
How to use checksums
31d85f95826528275335aec51646ca91e7bc767101021ef467ef7e5bf15d8506
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 Oct 23, 2026.

Transparency log

Release files / dml_rs-0.5.0-cp39-abi3-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

Download URL dml_rs-0.5.0-cp39-abi3-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
Size 735.1 kB
Tags CPython 3.9 Linux glibc 2.17+ ARM64 abi3
SHA-256 checksum
How to use checksums
7a2847524e572e613cb4944a67f1a1e1d54a83e8f26dfd813103f788ef862ab3
BLAKE2b-256 checksum
How to use checksums
7f425d391cb51168916180aca04e500cac1c036238ff8ee28bf854632d77df77
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 Oct 23, 2026.

Transparency log

Release files / dml_rs-0.5.0-cp39-abi3-macosx_11_0_arm64.whl

Download URL dml_rs-0.5.0-cp39-abi3-macosx_11_0_arm64.whl
Size 670.9 kB
Tags CPython 3.9 abi3 macOS 11.0+ ARM64
SHA-256 checksum
How to use checksums
f4148c9fccc6507f6e282b56cf8ef7a029139887401d55819aceb9a6e80ce153
BLAKE2b-256 checksum
How to use checksums
8d5f2da5cd777ed87b90302096129ac14100135ad73c2edd80061b3af4fb8299
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 Oct 23, 2026.

Transparency log

Release files / dml_rs-0.5.0-cp39-abi3-macosx_10_12_x86_64.whl

Download URL dml_rs-0.5.0-cp39-abi3-macosx_10_12_x86_64.whl
Size 711.7 kB
Tags CPython 3.9 abi3 macOS 10.12+ x86-64
SHA-256 checksum
How to use checksums
97dcfc474ed9289f08385f5598952e1735a204592dd5e2e2c1edb210a8cef2ab
BLAKE2b-256 checksum
How to use checksums
b3d99d03306b2ae20116523dfc11630b6b4f44336ff094ba1758d7e702c76acc
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 Oct 23, 2026.

Transparency log

Security for dml-rs 0.5.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.

Select from 6 files

Attestation for dml_rs-0.5.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 Oct 23, 2026.

Transparency log

A 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.

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page