diff-diff 3.6.2__tar.gz → 3.8.0__tar.gz

{diff_diff-3.6.2 → diff_diff-3.8.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: diff-diff
-Version: 3.6.2
+Version: 3.8.0
 Classifier: Development Status :: 5 - Production/Stable
 Classifier: Intended Audience :: Science/Research
 Classifier: Operating System :: OS Independent
@@ -20,9 +20,9 @@ Requires-Dist: scipy>=1.10
 Requires-Dist: pytest>=7.0 ; extra == 'dev'
 Requires-Dist: pytest-xdist>=3.0 ; extra == 'dev'
 Requires-Dist: pytest-cov>=4.0 ; extra == 'dev'
-Requires-Dist: black>=23.0 ; extra == 'dev'
-Requires-Dist: ruff>=0.1.0 ; extra == 'dev'
-Requires-Dist: mypy>=1.0 ; extra == 'dev'
+Requires-Dist: black==26.3.1 ; extra == 'dev'
+Requires-Dist: ruff==0.15.13 ; extra == 'dev'
+Requires-Dist: mypy==2.1.0 ; extra == 'dev'
 Requires-Dist: maturin>=1.4,<2.0 ; extra == 'dev'
 Requires-Dist: matplotlib>=3.5 ; extra == 'dev'
 Requires-Dist: nbmake>=1.5 ; extra == 'dev'
@@ -32,6 +32,7 @@ Requires-Dist: pydata-sphinx-theme>=0.16.1 ; extra == 'docs'
 Requires-Dist: sphinxext-opengraph>=0.9 ; extra == 'docs'
 Requires-Dist: sphinx-sitemap>=2.5 ; extra == 'docs'
 Requires-Dist: nbsphinx>=0.9 ; extra == 'docs'
+Requires-Dist: myst-parser>=2.0 ; extra == 'docs'
 Requires-Dist: matplotlib>=3.5 ; extra == 'docs'
 Requires-Dist: ipython>=8.0 ; extra == 'docs'
 Requires-Dist: plotly>=5.0 ; extra == 'plotly'
@@ -165,12 +166,14 @@ Full guide: `diff_diff.get_llm_guide("practitioner")`.
 - [TripleDifference](https://diff-diff.readthedocs.io/en/stable/api/triple_diff.html) - triple difference (DDD) estimator for designs requiring two criteria for treatment eligibility
 - [ContinuousDiD](https://diff-diff.readthedocs.io/en/stable/api/continuous_did.html) - Callaway, Goodman-Bacon & Sant'Anna (2024) continuous treatment DiD with dose-response curves
 - [HeterogeneousAdoptionDiD](https://diff-diff.readthedocs.io/en/stable/api/had.html) - de Chaisemartin, Ciccia, D'Haultfœuille & Knau (2026) for designs where **no unit remains untreated**; local-linear estimator at the dose support boundary returning Weighted Average Slope (WAS) on Design 1' (`d̲ = 0` / QUG) or `WAS_{d̲}` on Design 1 (`d̲ > 0`, continuous-near-d̲ or mass-point), with a multi-period event-study extension (last-treatment cohort, pointwise CIs). **Panel-only** in this release - repeated cross-sections rejected by the validator. Alias `HAD`.
+- [RegressionDiscontinuity](https://diff-diff.readthedocs.io/en/stable/api/regression_discontinuity.html) - Calonico, Cattaneo & Titiunik (2014) sharp, fuzzy, AND covariate-adjusted regression discontinuity with robust bias-corrected inference and rdrobust-parity bandwidth selection (all 10 selectors, mass-point handling; fuzzy via `treatment_col=` with a first-stage block and weak-identification warning; covariates via `covariates=` - CCFT 2019, same estimand, covariate-aware bandwidths). Canonical `att` is the bias-corrected estimate with a coherent robust CI (rdrobust's printed headline is `att_conventional`). Alias `RDD`.
 - [StackedDiD](https://diff-diff.readthedocs.io/en/stable/api/stacked_did.html) - Wing, Freedman & Hollingsworth (2024) stacked DiD with Q-weights and sub-experiments; optional covariate balancing (Ustyuzhanin 2026)
 - [EfficientDiD](https://diff-diff.readthedocs.io/en/stable/api/efficient_did.html) - Chen, Sant'Anna & Xie (2025) efficient DiD with optimal weighting for tighter SEs
 - [TROP](https://diff-diff.readthedocs.io/en/stable/api/trop.html) - Triply Robust Panel estimator (Athey et al. 2025) with nuclear norm factor adjustment
 - [StaggeredTripleDifference](https://diff-diff.readthedocs.io/en/stable/api/staggered.html#staggeredtripledifference) - Ortiz-Villavicencio & Sant'Anna (2025) staggered DDD with group-time ATT
 - [WooldridgeDiD](https://diff-diff.readthedocs.io/en/stable/api/wooldridge_etwfe.html) - Wooldridge (2023, 2025) ETWFE: saturated OLS, logit/Poisson QMLE (ASF-based ATT). Alias `ETWFE`.
 - [LPDiD](https://diff-diff.readthedocs.io/en/stable/api/lpdid.html) - Dube, Girardi, Jorda & Taylor (2025) Local Projections DiD: per-horizon long-difference event study on clean controls (no negative weighting), variance- or equally-weighted ATT, for absorbing or non-absorbing (reversible) treatment
+- [ChangesInChanges](https://diff-diff.readthedocs.io/en/stable/api/changes_in_changes.html) - Athey & Imbens (2006) nonlinear/distributional DiD for the 2x2 design: full counterfactual distribution and quantile treatment effects via CDF transformation, plus the QDiD comparison estimator; bootstrap inference; R qte parity. Alias `CiC`
 - [BaconDecomposition](https://diff-diff.readthedocs.io/en/stable/api/bacon.html) - Goodman-Bacon (2021) decomposition for diagnosing TWFE bias in staggered settings
 
 ## Diagnostics & Sensitivity
@@ -190,6 +193,7 @@ Most estimators accept an optional `survey_design` parameter (or `survey=` / `we
 - **Variance methods**: Taylor Series Linearization (TSL via Binder 1983), replicate weights (BRR / Fay / JK1 / JKn / SDR), survey-aware bootstrap
 - **Diagnostics**: DEFF per coefficient, effective n, subpopulation analysis, weight trimming, CV on estimates
 - **Repeated cross-sections**: `CallawaySantAnna(panel=False)` for BRFSS, ACS, CPS
+- **Weight calibration / raking**: upstream by design - pair with Meta's [balance](https://import-balance.org/) package, whose `balance.interop.diff_diff` adapter hands raked samples straight to diff-diff; see the [composition-drift tutorial](https://diff-diff.readthedocs.io/en/stable/tutorials/26_composition_drift_calibration.html)
 
 No other Python or R DiD package offers design-based variance estimation for modern heterogeneity-robust estimators.
 

{diff_diff-3.6.2 → diff_diff-3.8.0}/README.md

@@ -112,12 +112,14 @@ Full guide: `diff_diff.get_llm_guide("practitioner")`.
 - [TripleDifference](https://diff-diff.readthedocs.io/en/stable/api/triple_diff.html) - triple difference (DDD) estimator for designs requiring two criteria for treatment eligibility
 - [ContinuousDiD](https://diff-diff.readthedocs.io/en/stable/api/continuous_did.html) - Callaway, Goodman-Bacon & Sant'Anna (2024) continuous treatment DiD with dose-response curves
 - [HeterogeneousAdoptionDiD](https://diff-diff.readthedocs.io/en/stable/api/had.html) - de Chaisemartin, Ciccia, D'Haultfœuille & Knau (2026) for designs where **no unit remains untreated**; local-linear estimator at the dose support boundary returning Weighted Average Slope (WAS) on Design 1' (`d̲ = 0` / QUG) or `WAS_{d̲}` on Design 1 (`d̲ > 0`, continuous-near-d̲ or mass-point), with a multi-period event-study extension (last-treatment cohort, pointwise CIs). **Panel-only** in this release - repeated cross-sections rejected by the validator. Alias `HAD`.
+- [RegressionDiscontinuity](https://diff-diff.readthedocs.io/en/stable/api/regression_discontinuity.html) - Calonico, Cattaneo & Titiunik (2014) sharp, fuzzy, AND covariate-adjusted regression discontinuity with robust bias-corrected inference and rdrobust-parity bandwidth selection (all 10 selectors, mass-point handling; fuzzy via `treatment_col=` with a first-stage block and weak-identification warning; covariates via `covariates=` - CCFT 2019, same estimand, covariate-aware bandwidths). Canonical `att` is the bias-corrected estimate with a coherent robust CI (rdrobust's printed headline is `att_conventional`). Alias `RDD`.
 - [StackedDiD](https://diff-diff.readthedocs.io/en/stable/api/stacked_did.html) - Wing, Freedman & Hollingsworth (2024) stacked DiD with Q-weights and sub-experiments; optional covariate balancing (Ustyuzhanin 2026)
 - [EfficientDiD](https://diff-diff.readthedocs.io/en/stable/api/efficient_did.html) - Chen, Sant'Anna & Xie (2025) efficient DiD with optimal weighting for tighter SEs
 - [TROP](https://diff-diff.readthedocs.io/en/stable/api/trop.html) - Triply Robust Panel estimator (Athey et al. 2025) with nuclear norm factor adjustment
 - [StaggeredTripleDifference](https://diff-diff.readthedocs.io/en/stable/api/staggered.html#staggeredtripledifference) - Ortiz-Villavicencio & Sant'Anna (2025) staggered DDD with group-time ATT
 - [WooldridgeDiD](https://diff-diff.readthedocs.io/en/stable/api/wooldridge_etwfe.html) - Wooldridge (2023, 2025) ETWFE: saturated OLS, logit/Poisson QMLE (ASF-based ATT). Alias `ETWFE`.
 - [LPDiD](https://diff-diff.readthedocs.io/en/stable/api/lpdid.html) - Dube, Girardi, Jorda & Taylor (2025) Local Projections DiD: per-horizon long-difference event study on clean controls (no negative weighting), variance- or equally-weighted ATT, for absorbing or non-absorbing (reversible) treatment
+- [ChangesInChanges](https://diff-diff.readthedocs.io/en/stable/api/changes_in_changes.html) - Athey & Imbens (2006) nonlinear/distributional DiD for the 2x2 design: full counterfactual distribution and quantile treatment effects via CDF transformation, plus the QDiD comparison estimator; bootstrap inference; R qte parity. Alias `CiC`
 - [BaconDecomposition](https://diff-diff.readthedocs.io/en/stable/api/bacon.html) - Goodman-Bacon (2021) decomposition for diagnosing TWFE bias in staggered settings
 
 ## Diagnostics & Sensitivity
@@ -137,6 +139,7 @@ Most estimators accept an optional `survey_design` parameter (or `survey=` / `we
 - **Variance methods**: Taylor Series Linearization (TSL via Binder 1983), replicate weights (BRR / Fay / JK1 / JKn / SDR), survey-aware bootstrap
 - **Diagnostics**: DEFF per coefficient, effective n, subpopulation analysis, weight trimming, CV on estimates
 - **Repeated cross-sections**: `CallawaySantAnna(panel=False)` for BRFSS, ACS, CPS
+- **Weight calibration / raking**: upstream by design - pair with Meta's [balance](https://import-balance.org/) package, whose `balance.interop.diff_diff` adapter hands raked samples straight to diff-diff; see the [composition-drift tutorial](https://diff-diff.readthedocs.io/en/stable/tutorials/26_composition_drift_calibration.html)
 
 No other Python or R DiD package offers design-based variance estimation for modern heterogeneity-robust estimators.
 

{diff_diff-3.6.2 → diff_diff-3.8.0}/diff_diff/__init__.py

@@ -28,13 +28,58 @@ from diff_diff._backend import (
     _rust_project_simplex,
     _rust_solve_ols,
 )
-
+from diff_diff._guides_api import get_llm_guide
+from diff_diff.agent_workflow import agent_workflow
 from diff_diff.bacon import (
     BaconDecomposition,
     BaconDecompositionResults,
     Comparison2x2,
     bacon_decompose,
 )
+from diff_diff.business_report import (
+    BUSINESS_REPORT_SCHEMA_VERSION,
+    BusinessContext,
+    BusinessReport,
+)
+from diff_diff.chaisemartin_dhaultfoeuille import (
+    ChaisemartinDHaultfoeuille,
+    TWFEWeightsResult,
+    chaisemartin_dhaultfoeuille,
+    twowayfeweights,
+)
+from diff_diff.chaisemartin_dhaultfoeuille_results import (
+    ChaisemartinDHaultfoeuilleResults,
+    DCDHBootstrapResults,
+)
+from diff_diff.changes_in_changes import (
+    ChangesInChanges,
+    QDiD,
+)
+from diff_diff.changes_in_changes_results import (
+    ChangesInChangesResults,
+    QDiDResults,
+)
+from diff_diff.continuous_did import (
+    ContinuousDiD,
+    ContinuousDiDResults,
+    DoseResponseCurve,
+)
+from diff_diff.datasets import (
+    clear_cache,
+    list_datasets,
+    load_card_krueger,
+    load_castle_doctrine,
+    load_dataset,
+    load_divorce_laws,
+    load_mpdta,
+    load_prop99,
+    load_walmart,
+)
+from diff_diff.diagnostic_report import (
+    DIAGNOSTIC_REPORT_SCHEMA_VERSION,
+    DiagnosticReport,
+    DiagnosticReportResults,
+)
 from diff_diff.diagnostics import (
     PlaceboTestResults,
     leave_one_out_test,
@@ -44,22 +89,16 @@ from diff_diff.diagnostics import (
     run_all_placebo_tests,
     run_placebo_test,
 )
-from diff_diff.linalg import (
-    InferenceResult,
-    LinearRegression,
+from diff_diff.efficient_did import (
+    EDiDBootstrapResults,
+    EfficientDiD,
+    EfficientDiDResults,
 )
-from diff_diff.local_linear import (
-    KERNELS,
-    BandwidthResult,
-    BiasCorrectedFit,
-    LocalLinearFit,
-    bias_corrected_local_linear,
-    epanechnikov_kernel,
-    kernel_moments,
-    local_linear_fit,
-    mse_optimal_bandwidth,
-    triangular_kernel,
-    uniform_kernel,
+from diff_diff.estimators import (
+    DifferenceInDifferences,
+    MultiPeriodDiD,
+    SyntheticDiD,
+    TwoWayFixedEffects,
 )
 from diff_diff.had import (
     HeterogeneousAdoptionDiD,
@@ -80,12 +119,6 @@ from diff_diff.had_pretests import (
     stute_test,
     yatchew_hr_test,
 )
-from diff_diff.estimators import (
-    DifferenceInDifferences,
-    MultiPeriodDiD,
-    SyntheticDiD,
-    TwoWayFixedEffects,
-)
 from diff_diff.honest_did import (
     DeltaRM,
     DeltaSD,
@@ -96,6 +129,31 @@ from diff_diff.honest_did import (
     compute_honest_did,
     sensitivity_plot,
 )
+from diff_diff.imputation import (
+    ImputationBootstrapResults,
+    ImputationDiD,
+    ImputationDiDResults,
+    imputation_did,
+)
+from diff_diff.linalg import (
+    InferenceResult,
+    LinearRegression,
+)
+from diff_diff.local_linear import (
+    KERNELS,
+    BandwidthResult,
+    BiasCorrectedFit,
+    LocalLinearFit,
+    bias_corrected_local_linear,
+    epanechnikov_kernel,
+    kernel_moments,
+    local_linear_fit,
+    mse_optimal_bandwidth,
+    triangular_kernel,
+    uniform_kernel,
+)
+from diff_diff.lpdid import LPDiD
+from diff_diff.lpdid_results import LPDiDResults
 from diff_diff.power import (
     PowerAnalysis,
     PowerResults,
@@ -110,22 +168,16 @@ from diff_diff.power import (
     simulate_power,
     simulate_sample_size,
 )
-from diff_diff.pretrends import (
-    PreTrendsPower,
-    PreTrendsPowerCurve,
-    PreTrendsPowerResults,
-    compute_mdv,
-    compute_pretrends_power,
-)
+from diff_diff.practitioner import practitioner_next_steps
 from diff_diff.prep import (
     aggregate_survey,
     aggregate_to_cohorts,
     balance_panel,
     create_event_time,
     generate_continuous_did_data,
-    generate_did_data,
     generate_ddd_data,
     generate_ddd_panel_data,
+    generate_did_data,
     generate_event_study_data,
     generate_factor_data,
     generate_panel_data,
@@ -142,55 +194,44 @@ from diff_diff.prep import (
     validate_did_data,
     wide_to_long,
 )
+from diff_diff.pretrends import (
+    PreTrendsPower,
+    PreTrendsPowerCurve,
+    PreTrendsPowerResults,
+    compute_mdv,
+    compute_pretrends_power,
+)
+from diff_diff.profile import (
+    Alert,
+    OutcomeShape,
+    PanelProfile,
+    TreatmentDoseShape,
+    profile_panel,
+)
+from diff_diff.rdd import (
+    RegressionDiscontinuity,
+    RegressionDiscontinuityResults,
+)
 from diff_diff.results import (
     DiDResults,
     MultiPeriodDiDResults,
     PeriodEffect,
+    SpilloverDiDResults,  # re-export
     SyntheticDiDResults,
 )
-from diff_diff.survey import (
-    DEFFDiagnostics,
-    SurveyDesign,
-    SurveyMetadata,
-    compute_deff_diagnostics,
-    make_pweight_design,
-)
-from diff_diff.staggered import (
-    CallawaySantAnna,
-    CallawaySantAnnaResults,
-    CSBootstrapResults,
-    GroupTimeEffect,
-)
-from diff_diff.imputation import (
-    ImputationBootstrapResults,
-    ImputationDiD,
-    ImputationDiDResults,
-    imputation_did,
-)
-from diff_diff.two_stage import (
-    TwoStageBootstrapResults,
-    TwoStageDiD,
-    TwoStageDiDResults,
-    two_stage_did,
-)
 from diff_diff.spillover import (
     SpilloverDiD,
 )
-from diff_diff.results import SpilloverDiDResults  # re-export
 from diff_diff.stacked_did import (
     StackedDiD,
     StackedDiDResults,
     stacked_did,
 )
-from diff_diff.sun_abraham import (
-    SABootstrapResults,
-    SunAbraham,
-    SunAbrahamResults,
-)
-from diff_diff.triple_diff import (
-    TripleDifference,
-    TripleDifferenceResults,
-    triple_difference,
+from diff_diff.staggered import (
+    CallawaySantAnna,
+    CallawaySantAnnaResults,
+    CSBootstrapResults,
+    GroupTimeEffect,
 )
 from diff_diff.staggered_triple_diff import (
     StaggeredTripleDifference,
@@ -198,40 +239,39 @@ from diff_diff.staggered_triple_diff import (
 from diff_diff.staggered_triple_diff_results import (
     StaggeredTripleDiffResults,
 )
-from diff_diff.continuous_did import (
-    ContinuousDiD,
-    ContinuousDiDResults,
-    DoseResponseCurve,
+from diff_diff.sun_abraham import (
+    SABootstrapResults,
+    SunAbraham,
+    SunAbrahamResults,
 )
-from diff_diff.efficient_did import (
-    EfficientDiD,
-    EfficientDiDResults,
-    EDiDBootstrapResults,
+from diff_diff.survey import (
+    DEFFDiagnostics,
+    SurveyDesign,
+    SurveyMetadata,
+    compute_deff_diagnostics,
+    make_pweight_design,
 )
-from diff_diff.chaisemartin_dhaultfoeuille import (
-    ChaisemartinDHaultfoeuille,
-    TWFEWeightsResult,
-    chaisemartin_dhaultfoeuille,
-    twowayfeweights,
+from diff_diff.synthetic_control import (
+    SyntheticControl,
+    synthetic_control,
 )
-from diff_diff.chaisemartin_dhaultfoeuille_results import (
-    ChaisemartinDHaultfoeuilleResults,
-    DCDHBootstrapResults,
+from diff_diff.synthetic_control_results import SyntheticControlResults
+from diff_diff.triple_diff import (
+    TripleDifference,
+    TripleDifferenceResults,
+    triple_difference,
 )
 from diff_diff.trop import (
     TROP,
     TROPResults,
     trop,
 )
-from diff_diff.synthetic_control import (
-    SyntheticControl,
-    synthetic_control,
+from diff_diff.two_stage import (
+    TwoStageBootstrapResults,
+    TwoStageDiD,
+    TwoStageDiDResults,
+    two_stage_did,
 )
-from diff_diff.synthetic_control_results import SyntheticControlResults
-from diff_diff.wooldridge import WooldridgeDiD
-from diff_diff.wooldridge_results import WooldridgeDiDResults
-from diff_diff.lpdid import LPDiD
-from diff_diff.lpdid_results import LPDiDResults
 from diff_diff.utils import (
     WildBootstrapResults,
     check_parallel_trends,
@@ -252,35 +292,8 @@ from diff_diff.visualization import (
     plot_staircase,
     plot_synth_weights,
 )
-from diff_diff.practitioner import practitioner_next_steps
-from diff_diff.business_report import (
-    BUSINESS_REPORT_SCHEMA_VERSION,
-    BusinessContext,
-    BusinessReport,
-)
-from diff_diff.diagnostic_report import (
-    DIAGNOSTIC_REPORT_SCHEMA_VERSION,
-    DiagnosticReport,
-    DiagnosticReportResults,
-)
-from diff_diff._guides_api import get_llm_guide
-from diff_diff.agent_workflow import agent_workflow
-from diff_diff.profile import (
-    Alert,
-    OutcomeShape,
-    PanelProfile,
-    TreatmentDoseShape,
-    profile_panel,
-)
-from diff_diff.datasets import (
-    clear_cache,
-    list_datasets,
-    load_card_krueger,
-    load_castle_doctrine,
-    load_dataset,
-    load_divorce_laws,
-    load_mpdta,
-)
+from diff_diff.wooldridge import WooldridgeDiD
+from diff_diff.wooldridge_results import WooldridgeDiDResults
 
 # Estimator aliases — short names for convenience
 DiD = DifferenceInDifferences
@@ -300,8 +313,10 @@ EDiD = EfficientDiD
 ETWFE = WooldridgeDiD
 DCDH = ChaisemartinDHaultfoeuille
 HAD = HeterogeneousAdoptionDiD
+CiC = ChangesInChanges
+RDD = RegressionDiscontinuity
 
-__version__ = "3.6.2"
+__version__ = "3.8.0"
 __all__ = [
     # Estimators
     "DifferenceInDifferences",
@@ -310,6 +325,8 @@ __all__ = [
     "SyntheticDiD",
     "CallawaySantAnna",
     "ChaisemartinDHaultfoeuille",
+    "ChangesInChanges",
+    "QDiD",
     "ContinuousDiD",
     "SunAbraham",
     "ImputationDiD",
@@ -327,6 +344,7 @@ __all__ = [
     "CS",
     "CDiD",
     "DCDH",
+    "CiC",
     "SA",
     "BJS",
     "Gardner",
@@ -347,6 +365,8 @@ __all__ = [
     "CallawaySantAnnaResults",
     "CSBootstrapResults",
     "GroupTimeEffect",
+    "ChangesInChangesResults",
+    "QDiDResults",
     "ContinuousDiDResults",
     "DoseResponseCurve",
     "SunAbrahamResults",
@@ -495,6 +515,10 @@ __all__ = [
     "HeterogeneousAdoptionDiDResults",
     "HeterogeneousAdoptionDiDEventStudyResults",
     "HAD",
+    # RegressionDiscontinuity (sharp RD, rdrobust parity)
+    "RegressionDiscontinuity",
+    "RegressionDiscontinuityResults",
+    "RDD",
     # HeterogeneousAdoptionDiD pre-test diagnostics (Phase 3)
     "qug_test",
     "stute_test",
@@ -515,6 +539,8 @@ __all__ = [
     "load_castle_doctrine",
     "load_divorce_laws",
     "load_mpdta",
+    "load_prop99",
+    "load_walmart",
     "load_dataset",
     "list_datasets",
     "clear_cache",

{diff_diff-3.6.2 → diff_diff-3.8.0}/diff_diff/_backend.py

@@ -18,27 +18,59 @@ _backend_env = os.environ.get("DIFF_DIFF_BACKEND", "auto").lower()
 # Try to import Rust backend for accelerated operations
 try:
     from diff_diff._rust_backend import (
-        generate_bootstrap_weights_batch as _rust_bootstrap_weights,
-        project_simplex as _rust_project_simplex,
-        solve_ols as _rust_solve_ols,
+        bootstrap_trop_variance as _rust_bootstrap_trop_variance,
+    )
+    from diff_diff._rust_backend import (
+        bootstrap_trop_variance_global as _rust_bootstrap_trop_variance_global,
+    )
+    from diff_diff._rust_backend import (
+        compute_noise_level as _rust_compute_noise_level,
+    )
+    from diff_diff._rust_backend import (
         compute_robust_vcov as _rust_compute_robust_vcov,
+    )
+    from diff_diff._rust_backend import (
+        # SDID weights (Frank-Wolfe matching R's synthdid)
+        compute_sdid_unit_weights as _rust_sdid_unit_weights,
+    )
+    from diff_diff._rust_backend import (
+        compute_time_weights as _rust_compute_time_weights,
+    )
+    from diff_diff._rust_backend import (
         # TROP estimator acceleration (local method)
         compute_unit_distance_matrix as _rust_unit_distance_matrix,
+    )
+    from diff_diff._rust_backend import (
+        generate_bootstrap_weights_batch as _rust_bootstrap_weights,
+    )
+    from diff_diff._rust_backend import (
         loocv_grid_search as _rust_loocv_grid_search,
-        bootstrap_trop_variance as _rust_bootstrap_trop_variance,
+    )
+    from diff_diff._rust_backend import (
         # TROP estimator acceleration (global method)
         loocv_grid_search_global as _rust_loocv_grid_search_global,
-        bootstrap_trop_variance_global as _rust_bootstrap_trop_variance_global,
-        # SDID weights (Frank-Wolfe matching R's synthdid)
-        compute_sdid_unit_weights as _rust_sdid_unit_weights,
-        compute_time_weights as _rust_compute_time_weights,
-        compute_noise_level as _rust_compute_noise_level,
+    )
+    from diff_diff._rust_backend import (
+        project_simplex as _rust_project_simplex,
+    )
+    from diff_diff._rust_backend import (
+        # Diagnostics
+        rust_backend_info as _rust_backend_info,
+    )
+    from diff_diff._rust_backend import (
         sc_weight_fw as _rust_sc_weight_fw,
-        sc_weight_fw_with_convergence as _rust_sc_weight_fw_with_convergence,
+    )
+    from diff_diff._rust_backend import (
         sc_weight_fw_weighted as _rust_sc_weight_fw_weighted,
+    )
+    from diff_diff._rust_backend import (
         sc_weight_fw_weighted_with_convergence as _rust_sc_weight_fw_weighted_with_convergence,
-        # Diagnostics
-        rust_backend_info as _rust_backend_info,
+    )
+    from diff_diff._rust_backend import (
+        sc_weight_fw_with_convergence as _rust_sc_weight_fw_with_convergence,
+    )
+    from diff_diff._rust_backend import (
+        solve_ols as _rust_solve_ols,
     )
 
     _rust_available = True
@@ -73,6 +105,37 @@ try:
 except ImportError:
     _rust_demean_map = None
 
+# Batched ridge-regularized SPD solve (EfficientDiD per-unit weights):
+# imported independently for the same mixed-version reason as demean_map.
+try:
+    from diff_diff._rust_backend import (
+        batched_ridge_chol_solve_ones as _rust_batched_ridge_chol_solve,
+    )
+except ImportError:
+    _rust_batched_ridge_chol_solve = None
+
+# HC2 (leverage-corrected) robust vcov: imported independently for the same
+# mixed-version reason as demean_map (a stale extension missing only this
+# newer symbol degrades HC2 to the NumPy path without disabling the older
+# Rust accelerations).
+try:
+    from diff_diff._rust_backend import (
+        compute_robust_vcov_hc2 as _rust_compute_robust_vcov_hc2,
+    )
+except ImportError:
+    _rust_compute_robust_vcov_hc2 = None
+
+# Opt-in normal-equations Cholesky OLS fast path: imported independently
+# for the same mixed-version reason as demean_map. A stale extension
+# missing only this symbol keeps every older Rust acceleration: Rust-eligible
+# fits fall back to the legacy SVD solve_ols kernel (the knob simply has no
+# Rust acceleration there), while numpy-lane fits (weighted, non-hc1,
+# forced-python) still use the numpy Cholesky twin.
+try:
+    from diff_diff._rust_backend import solve_ols_chol as _rust_solve_ols_chol
+except ImportError:
+    _rust_solve_ols_chol = None
+
 # Determine final backend based on environment variable and availability
 if _backend_env == "python":
     # Force pure Python mode - disable Rust even if available
@@ -83,6 +146,12 @@ if _backend_env == "python":
     _rust_compute_robust_vcov = None
     # FE-absorption MAP demeaning kernel
     _rust_demean_map = None
+    # Batched ridge-regularized SPD solve
+    _rust_batched_ridge_chol_solve = None
+    # HC2 robust vcov
+    _rust_compute_robust_vcov_hc2 = None
+    # Opt-in normal-equations Cholesky OLS fast path
+    _rust_solve_ols_chol = None
     # TROP estimator acceleration (local method)
     _rust_unit_distance_matrix = None
     _rust_loocv_grid_search = None
@@ -134,8 +203,12 @@ __all__ = [
     "_rust_project_simplex",
     "_rust_solve_ols",
     "_rust_compute_robust_vcov",
+    # Opt-in normal-equations Cholesky OLS fast path
+    "_rust_solve_ols_chol",
     # FE-absorption MAP demeaning kernel
     "_rust_demean_map",
+    # Batched ridge-regularized SPD solve (EfficientDiD per-unit weights)
+    "_rust_batched_ridge_chol_solve",
     # TROP estimator acceleration (local method)
     "_rust_unit_distance_matrix",
     "_rust_loocv_grid_search",