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.2.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.2.0
File Size Uploaded Provenance
dml_rs-0.2.0.tar.gz 76.5 kB With provenance Details

Built distributions (wheels)

Table of built distributions (wheels) for dml-rs 0.2.0
File
dml_rs-0.2.0-cp39-abi3-win_amd64.whl CPython 3.9 abi3 Windows x86-64 Details
dml_rs-0.2.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.2.0-cp39-abi3-manylinux_2_17_aarch64.manylinux2014_aarch64.whl CPython 3.9 abi3 Linux glibc 2.17+ ARM64 Details
dml_rs-0.2.0-cp39-abi3-macosx_11_0_arm64.whl CPython 3.9 abi3 macOS 11.0+ ARM64 Details
dml_rs-0.2.0-cp39-abi3-macosx_10_12_x86_64.whl CPython 3.9 abi3 macOS 10.12+ x86-64 Details

Total release size: 3.1 MB

Release files / dml_rs-0.2.0.tar.gz

Download URL dml_rs-0.2.0.tar.gz
Size 76.5 kB
Tags Source
SHA-256 checksum
How to use checksums
a203863913ddac14aa8bfc25f2cce184d5a7c1cb059ed786fb1bba2b9b49a1ed
BLAKE2b-256 checksum
How to use checksums
2823d849a95943e8777296c0638d895bc94aae50140c9fbd49d7e610188b656e
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.2.0-cp39-abi3-win_amd64.whl

Download URL dml_rs-0.2.0-cp39-abi3-win_amd64.whl
Size 498.0 kB
Tags CPython 3.9 Windows x86-64 abi3
SHA-256 checksum
How to use checksums
592e4b6e8cd672e98848161b43bfdf0d29ee368100faf720ca440b45b4a3ae5d
BLAKE2b-256 checksum
How to use checksums
7b8e8d20932a296982290b7e9e3ea28a4f8745244c421b8bd8747c3e032b32c3
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.2.0-cp39-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

Download URL dml_rs-0.2.0-cp39-abi3-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Size 683.9 kB
Tags CPython 3.9 Linux glibc 2.17+ x86-64 abi3
SHA-256 checksum
How to use checksums
a3a31b56c6effc073d58b446d4af58bd6149f3663640eb2d48006b28991abaa9
BLAKE2b-256 checksum
How to use checksums
009d84db87a3ca0240d6bbc945e81601dcb3fa833e80557608480db69db619e6
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.2.0-cp39-abi3-manylinux_2_17_aarch64.manylinux2014_aarch64.whl

Download URL dml_rs-0.2.0-cp39-abi3-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
Size 651.5 kB
Tags CPython 3.9 Linux glibc 2.17+ ARM64 abi3
SHA-256 checksum
How to use checksums
4ac34681fe85b8d55339c877269619e583034ae79ac97023d2320d4a82f4de1b
BLAKE2b-256 checksum
How to use checksums
1d6be720db2d454f907cd47bd0b24d7705103434e4ff94370fd0ef6c959bfd68
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.2.0-cp39-abi3-macosx_11_0_arm64.whl

Download URL dml_rs-0.2.0-cp39-abi3-macosx_11_0_arm64.whl
Size 594.1 kB
Tags CPython 3.9 abi3 macOS 11.0+ ARM64
SHA-256 checksum
How to use checksums
31c841ef22adac546127223d78e6d58c32c0f3a572c3d1fc61e53fedd78a95c1
BLAKE2b-256 checksum
How to use checksums
c28bc562b89867b42ac8f876ed9c8183caaf782d967cddedc550154007a07927
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.2.0-cp39-abi3-macosx_10_12_x86_64.whl

Download URL dml_rs-0.2.0-cp39-abi3-macosx_10_12_x86_64.whl
Size 623.4 kB
Tags CPython 3.9 abi3 macOS 10.12+ x86-64
SHA-256 checksum
How to use checksums
9b6fef5ab7978a0a379965b6fbb858be176d154f64da081177a68634f4db279e
BLAKE2b-256 checksum
How to use checksums
b8a8f9118a800dd165970bd260a3156ea2da0a0aae405ea7d5bbb5972fc55dc5
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.2.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.2.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