Packages
527 packages for ML, causal inference, time series, and more.
Staggered Difference-in-Differences
5 packages
csdid
Python adaptation of the R `did` package. Implements multi-period DiD with staggered treatment timing (Callaway & …

bacondecomp
Performs Goodman-Bacon decomposition showing how two-way fixed effects (TWFE) estimates are weighted averages of all …

did
Implements group-time average treatment effects (ATT(g,t)) for staggered DiD designs with multiple periods and variation …

staggered
Provides the efficient estimator for randomized staggered rollout designs, offering optimal weighting schemes for …

Differences
Implements modern difference-in-differences methods for staggered adoption designs (e.g., Callaway & Sant'Anna).
Synthetic Control Methods
15 packagesscpi
Provides rigorous prediction intervals for synthetic control methods following Cattaneo et al. (2021, 2025). Supports …
microsynth
Extends synthetic control method to micro-level data with many units. Implements permutation inference and handles …

augsynth
Implements the Augmented Synthetic Control Method, which uses an outcome model (ridge regression by default) to correct …
SCtools
Automates placebo tests and multi-treated-unit ATT calculations for synthetic control. Provides utilities for generating …

mlsynth
Implements advanced synthetic control methods: forward DiD, cluster SC, factor models, and proximal SC. Designed for …

synthlearners
Fast synthetic control estimators for panel data problems. Optimized ATT estimation with multiple SC algorithms.

GeoLift
Meta's geo-experimental methodology combining Augmented Synthetic Control with power analysis

gsynth
Implements generalized synthetic control with interactive fixed effects, extending SCM to multiple treated units with …
tidysynth
Brings synthetic control method into the tidyverse with cleaner syntax and built-in placebo inference. Provides …

synthdid
Implements synthetic difference-in-differences, a hybrid method combining insights from both DiD and synthetic control …
DiD Extensions & Variants
15 packagespycinc
Changes‑in‑Changes (CiC) estimator for distributional treatment effects (Athey & Imbens 2006).

DRDID
Implements locally efficient doubly robust DiD estimators that combine inverse probability weighting and outcome …

trimmed_match
Google's robust analysis for paired geo experiments using trimmed statistics. Handles outliers in geo-level data.

matched_markets
Google's time-based regression with greedy search for optimal geo experiment groups.

didet
DiD with general treatment patterns. Handles effective treatment timing beyond simple staggered adoption.

TFP CausalImpact
TensorFlow Probability port of Google's CausalImpact. Bayesian structural time-series for intervention effects.
pyleebounds
Lee (2009) sample‑selection bounds for treatment effects; trims treated distribution to match selection rates.

didhetero
Doubly robust estimation for group-time conditional average treatment effects. UCB for heterogeneous DiD.

CausalImpact
Python port of Google's R package for estimating causal effects of interventions on time series using Bayesian …

HonestDiD
Constructs robust confidence intervals for DiD and event-study designs under violations of parallel trends. Allows …
Dynamic Structural Models
10 packages
dcegm
JAX-compatible DC-EGM algorithm for discrete-continuous dynamic programming (Iskhakov et al. 2017).

respy
Simulation and estimation of finite-horizon dynamic discrete choice (DDC) models (e.g., labor/education choice).

Dolo
Framework for describing and solving economic models (DSGE, OLG, etc.) using a declarative YAML-based format.

upper-envelope
Fast upper envelope scan for discrete-continuous dynamic programming. JAX and numba implementations.

gEconpy
DSGE modeling tools inspired by R's gEcon. Automatic first-order condition derivation with Dynare export.

pydsge
DSGE model simulation, filtering, and Bayesian estimation. Handles occasionally binding constraints.

pynare
Python wrapper/interface to Dynare for DSGE model solving. Bridge between Python workflows and Dynare.

econpizza
Solve nonlinear heterogeneous agent models (HANK) with perfect foresight. Efficient perturbation and projection methods.

ruspy
Python package for simulation and estimation of Rust (1987) bus engine replacement model. Implements the nested fixed …

QuantEcon.py
Core library for quantitative economics: dynamic programming, Markov chains, game theory, numerical methods.
Structural Econometrics
16 packages
Greeners
Comprehensive Rust econometrics library with OLS, IV, panel data estimators, fixed effects, DiD, and …

Biogeme
Maximum likelihood estimation of parametric models, with strong support for complex discrete choice models.

torch-choice
PyTorch framework for flexible estimation of complex discrete choice models, leveraging GPU acceleration.

jaxonometrics
JAX-ecosystem implementations of standard econometrics routines for GPU computation.

Argmin
Numerical optimization framework for Rust with Newton, BFGS, L-BFGS, trust region, and derivative-free methods for …

torchonometrics
Econometrics implementations in PyTorch. Leverages autodiff and GPU acceleration for econometric methods.

gegravity
General equilibrium structural gravity modeling for trade policy analysis. Only Python package for Anderson-van Wincoop …
momentfit
Modern S4-based implementation of Generalized Method of Moments supporting systems of equations, nonlinear moment …

PyLogit
Flexible implementation of conditional/multinomial logit models with utilities for data preparation.

OpenMx
Extended SEM software with programmatic model specification via paths (RAM) or matrix algebra, supporting mixture …
IV & RDD Packages
17 packagesbunching
Implements Kleven-Waseem style bunching estimation for kink and notch designs. Calculates parametric elasticities from …
ATbounds
Implements modern treatment effect bounds beyond basic Manski worst-case scenarios. Provides tighter bounds using …
latenetwork
Handles both noncompliance AND network interference of unknown form following Hoshino and Yanagi (2023 JASA). Provides …

hdm
High-dimensional statistical methods featuring heteroscedasticity-robust LASSO with theoretically-grounded penalty …

rddensity
Implements manipulation testing (density discontinuity testing) procedures using local polynomial density estimators to …
rddtools
Regression discontinuity design toolkit with clustered inference for geographic discontinuities. Provides bandwidth …
AER
Companion package to 'Applied Econometrics with R' (Kleiber & Zeileis) plus datasets from Stock & Watson. Provides …

rdpower
Provides tools for power, sample size, and minimum detectable effects (MDE) calculations in RD designs using robust …
Causal Inference Libraries
14 packages
DoubleML
Implements the double/debiased ML framework (Chernozhukov et al.) for estimating causal parameters (ATE, LATE, POM) with …

PySensemakr
Implements Cinelli-Hazlett framework for assessing robustness to unobserved confounding. Computes confounder strength …

scikit-uplift
Focuses on uplift modeling and estimating heterogeneous treatment effects using various ML-based methods.

pydtr
Dynamic treatment regimes using Iterative Q-Learning. Scikit-learn compatible for multi-stage optimal treatment …

DoWhy
End-to-end framework for causal inference based on causal graphs (DAGs) and potential outcomes. Covers identification, …

DoWhy
End-to-end framework for causal inference based on causal graphs (DAGs) and potential outcomes. Covers identification, …
Panel Data Methods
18 packages
Linearmodels
Estimation of fixed, random, pooled OLS models for panel data. Also Fama-MacBeth and between/first-difference …

FixedEffectModel
Panel data modeling with IV tests (weak IV, over-identification, endogeneity) and 2-step GMM estimation.

bife
Estimates fixed effects binary choice models (logit and probit) with potentially many individual fixed effects using a …

plm
Comprehensive econometrics package for linear panel models providing fixed effects (within), random effects, between, …

pydynpd
Estimation of dynamic panel data models using Arellano-Bond (Difference GMM) and Blundell-Bond (System GMM). Includes …
PyFixest
Fast estimation of linear models with multiple high-dimensional fixed effects (like R's `fixest`). Supports OLS, IV, …
duckreg
Out-of-core regression (OLS/IV) for very large datasets using DuckDB aggregation. Handles data that doesn't fit in …

glmmTMB
Fit generalized linear mixed models with extensions including zero-inflation, hurdle models, heteroscedasticity, and …
nlme
Fit Gaussian linear and nonlinear mixed-effects models with flexible correlation structures, variance functions for …
Causal Discovery
17 packages
causal-learn
Comprehensive Python package serving as Python translation and extension of Java-based Tetrad toolkit for causal …

Ananke
Causal inference using graphical models (DAGs), including identification theory and effect estimation.

MCD
Mixture of Causal Graphs discovery for heterogeneous time series (ICML 2024). Finds time-varying causal structures.
bnlearn
Bayesian network structure learning, parameter estimation, and inference. Implements constraint-based (PC, GS), …

dagitty
Analysis of structural causal models represented as DAGs. Computes adjustment sets, identifies instrumental variables, …

causal-llm-bfs
LLM + BFS hybrid for efficient causal graph discovery. Uses language models to guide structure search.

gCastle
Huawei Noah's Ark Lab end-to-end causal structure learning toolchain emphasizing gradient-based methods with GPU …

SDCI
State-dependent causal inference for conditionally stationary processes (ICML 2025). Handles regime-switching causal …

ggdag
Visualize and analyze causal DAGs using ggplot2. Provides tidy interface to dagitty with publication-quality DAG plots, …
pcalg
Causal structure learning from observational data using the PC algorithm and variants. Estimates Markov equivalence …
Matching & Weighting
15 packagesCausalMatch
Implements Propensity Score Matching (PSM) and Coarsened Exact Matching (CEM) with ML flexibility for propensity score …
CausalGPS
Machine learning-based generalized propensity score estimation for continuous treatments. Uses SuperLearner ensemble …
fastmatch
Fast k-nearest-neighbor matching for large datasets using Facebook's FAISS library.

CausalInference
Implements classical causal inference methods like propensity score matching, inverse probability weighting, …

MatchIt
Comprehensive matching package that selects matched samples of treated and control groups with similar covariate …

optmatch
Distance-based bipartite matching using minimum cost network flow algorithms, oriented to matching treatment and control …
ebal
Implements entropy balancing, a reweighting method that finds weights for control units such that specified covariate …

CBPS
Implements Covariate Balancing Propensity Score, which estimates propensity scores by jointly optimizing treatment …

WeightIt
Unified interface for generating balancing weights for causal effect estimation in observational studies. Supports …

cobalt
Generates standardized balance tables and plots for covariates after preprocessing via matching, weighting, or …
Mediation Analysis
10 packages
pyregadj
Regression and ML adjustments to treatment effects in RCTs. Implements List et al. (2024) methods.

inferference
Computes inverse probability weighted (IPW) causal effects under partial interference following Tchetgen Tchetgen and …

EValue
Conducts sensitivity analyses for unmeasured confounding, selection bias, and measurement error in observational studies …

causalweight
Semiparametric causal inference methods based on inverse probability weighting and double machine learning for average …

ddml
Streamlined double/debiased machine learning estimation with emphasis on (short-)stacking to combine multiple base …

sensemakr
Suite of sensitivity analysis tools extending the traditional omitted variable bias framework, computing robustness …
CMAverse
Unified interface for six causal mediation approaches including traditional regression, inverse odds weighting, and …
mediation
Estimates Average Causal Mediation Effects (ACME) with sensitivity analysis for unmeasured confounding. Implements …
PStrata
Principal stratification analysis for noncompliance and truncation-by-death using both Bayesian (Stan) and frequentist …

spilled_t
Treatment and spillover effect estimation under network interference. Separates direct and indirect effects.
CATE Estimation
10 packages
KECENI
Doubly robust, non-parametric estimation of node-wise counterfactual means under network interference (arXiv 2024).

EconML
Microsoft toolkit for estimating heterogeneous treatment effects using DML, causal forests, meta-learners, and …

CATENets
JAX-accelerated neural network CATE estimators implementing SNet, FlexTENet, TARNet, CFRNet, and DragonNet …

mcf (Modified Causal Forest)
Comprehensive Python implementation for heterogeneous treatment effect estimation. Handles binary/multiple discrete …

bartCause
Bayesian Additive Regression Trees for causal inference. Strong ACIC competition performer with sparsity-inducing priors …

CausalML
Focuses on uplift modeling and heterogeneous treatment effect estimation using machine learning techniques.

grf
Forest-based statistical estimation and inference for heterogeneous treatment effects, supporting multiple treatment …

causalToolbox
Implements meta-learner algorithms (S-learner, T-learner, X-learner) for heterogeneous treatment effect estimation using …
DynTxRegime
Comprehensive package for dynamic treatment regimes implementing Q-learning, value search, and outcome-weighted learning …
DTRreg
Dynamic treatment regime estimation via G-estimation for sequential treatment decisions. Implements methods for finding …
Targeted Learning
6 packages
aipyw
Minimal, fast AIPW (Augmented Inverse Probability Weighting) implementation for discrete treatments. Sklearn-compatible …

causal-curve
Continuous treatment dose-response curve estimation. GPS and TMLE methods for continuous treatments.

ltmle
Targeted maximum likelihood estimation for treatment/censoring-specific mean outcomes with time-varying treatments and …

SuperLearner
Implements the Super Learner algorithm for optimal ensemble prediction via cross-validation. Creates weighted …
tmle
Implements targeted maximum likelihood estimation for point treatment effects with binary or continuous outcomes. …

tmle3
A modular, extensible framework for targeted minimum loss-based estimation supporting custom TMLE parameters through a …
Data Visualization
7 packages
cowplot
Publication-ready ggplot2 themes and plot arrangement utilities. Provides clean themes, plot annotations, and functions …

patchwork
Compose multiple ggplot2 plots into publication-ready multi-panel figures. Uses intuitive operators (+, |, /) for …

see
Visualization toolbox for the easystats ecosystem built on ggplot2. Provides publication-ready plotting methods for …
Connected Papers
Visual tool for exploring academic paper relationships. Creates visual graphs showing prior and derivative works.
stargazer
Produces well-formatted LaTeX, HTML/CSS, and ASCII regression tables with multiple models side-by-side, plus summary …

gt
Build display tables from tabular data using a cohesive grammar of table parts (header, stub, body, footer). Enables …

modelsummary
Creates publication-quality tables summarizing multiple statistical models side-by-side, plus coefficient plots, data …
Data Manipulation
6 packages
collapse
High-performance data transformation package designed by an economist. Provides fast grouped operations, time series …

tidyverse
Meta-package installing core tidyverse packages: ggplot2 (visualization), dplyr (manipulation), tidyr (tidying), readr …

Polars
Blazingly fast DataFrame library for Rust and Python with SQL-like syntax, lazy evaluation, and excellent time series …

countrycode
R package for converting between country naming and coding conventions essential for merging defense datasets

data.table
Extension of data.frame providing fast aggregation of large data (100GB+), ordered joins, and memory-efficient …

haven
Import and export Stata, SPSS, and SAS data files preserving variable labels and value labels. Handles .dta, .sav, …
Power Analysis & Experimental Design
12 packages
DeclareDesign
Ex ante experimental design declaration and diagnosis. Enables researchers to formally describe their research design, …

fabricatr
Simulates realistic social science data for power analysis and design testing. Creates hierarchical data structures with …

Superpower
Simulation-based power analysis for factorial ANOVA designs (up to 3 factors). Includes Shiny app for interactive power …

ADOpy
Bayesian Adaptive Design Optimization (ADO) for tuning experiments in real-time, with models for psychometric tasks.

pyDOE2
Implements classical Design of Experiments: factorial (full/fractional), response surface (Box-Behnken, CCD), Latin …

WebPower
Comprehensive collection of tools for basic and advanced statistical power analysis including correlation, t-test, …

Adaptive
Parallel active learning library for adaptive function sampling/evaluation, with live plotting for monitoring.

DoEgen
Automates generation and optimization of designs, especially for mixed factor-level experiments; computes efficiency …

simChef
DGP (Data Generating Process) framework for systematic simulation studies. Enables reproducible computational …
mlpwr
Machine learning-based power analysis using surrogate models. Efficient sample size planning for complex study designs.
Bandit Algorithms
9 packages
Ax (Meta Adaptive Experimentation)
Meta's open-source platform for adaptive experimentation. Bayesian optimization, multi-objective optimization, and …

ContextualBandits
Implements a wide range of contextual bandit algorithms (linear, tree-based, neural) and off-policy evaluation methods.

BayesianBandits
Lightweight microframework for Bayesian bandits (Thompson Sampling) with support for contextual/restless/delayed …

MABWiser
Production-ready, scikit-learn style library for contextual & stochastic bandits with parallelism and simulation tools.

Open Bandit Pipeline (OBP)
Framework for **offline evaluation (OPE)** of bandit policies using logged data. Implements IPS, DR, DM estimators.
abracadabra
Sequential testing with always-valid inference. Supports continuous monitoring of A/B tests.

PyXAB
Library for advanced bandit problems: X-armed bandits (continuous/structured action spaces) and online optimization.

SMPyBandits
Comprehensive research framework for single/multi-player MAB algorithms (stochastic, adversarial, contextual).

contextual
Multi-armed bandit algorithms including Thompson Sampling, UCB, and LinUCB. Directly applicable to adaptive A/B testing …
Experiment Analysis
7 packages
randomizr
Proper randomization procedures for experiments with known assignment probabilities. Implements simple, complete, block, …
cjoint
Estimates Average Marginal Component Effects (AMCEs) for conjoint experiments following Hainmueller, Hopkins & Yamamoto …

cregg
Tidy interface for conjoint analysis with visualization. Provides functions for calculating and plotting marginal means …

tea-tasting
Calculate A/B test statistics directly within data warehouses (BigQuery, ClickHouse, Snowflake, Spark) via Ibis …

CausalMotifs
Meta's library for estimating heterogeneous spillover effects in A/B tests. Handles network interference.

savvi
Safe Anytime Valid Inference using e-processes and confidence sequences (Ramdas et al. 2023). Valid inference at any …

Ambrosia
End-to-end A/B testing from MobileTeleSystems with PySpark support. Covers experiment design, multi-group splitting, …
Actuarial Science
11 packages
actuar
Actuarial science functions for R including loss distributions, credibility theory, ruin theory, and simulation of …
extRemes
Comprehensive toolkit for extreme value analysis with diagnostic plots, return level estimation, and non-stationary …

chainladder-python
Python library for actuarial reserving implementing chain-ladder, Bornhuetter-Ferguson, Cape Cod, and stochastic methods …

Fairlearn
Microsoft toolkit for assessing and improving ML model fairness, critical for insurance pricing compliance and avoiding …

lifelib
Open-source actuarial library with complete life insurance projection models including term, whole life, universal life, …

ChainLadder
Comprehensive R package for claims reserving methods including Mack, Munich, and bootstrap chain-ladder with full …
Energy Systems
8 packages
eiapy
Python wrapper for the EIA Open Data API. Access generation, consumption, prices, and other energy data …

eiapy
Python wrapper for the EIA Open Data API. Access generation, consumption, prices, and other energy data …

gridstatus
Unified Python interface for U.S. electricity grid data from all major ISOs

OpenDSS
EPRI's open-source distribution system simulator. Quasi-static time-series analysis, DER integration, and comprehensive …

OpenDSS
EPRI's open-source distribution system simulator. Quasi-static time-series analysis, DER integration, and comprehensive …

pandapower
Power system analysis for distribution networks. Newton-Raphson power flow, state estimation, short circuit …

pandapower
Power system analysis for distribution networks. Newton-Raphson power flow, state estimation, short circuit …

GenX
Capacity expansion model from MIT/Princeton in Julia. Highly configurable with unit commitment, long-duration storage, …

GenX
Capacity expansion model from MIT/Princeton in Julia. Highly configurable with unit commitment, long-duration storage, …

PowerModels.jl
Power network optimization in Julia. Supports AC/DC optimal power flow, transmission expansion, and custom formulations …
Marketing Analytics
13 packages
LightweightMMM
Bayesian Marketing Mix Modeling (see Marketing Mix Models section).

PyMC Marketing
Collection of Bayesian marketing models built with PyMC, including MMM, CLV, and attribution.

PyMC-Marketing
Bayesian Marketing Mix Modeling and Customer Lifetime Value with PyMC, including GPU acceleration

PyMC-Marketing
Bayesian Marketing Mix Modeling and Customer Lifetime Value with PyMC, including GPU acceleration

mmm_stan
Python/STAN implementation of Bayesian Marketing Mix Models.

pylift
Wayfair's uplift modeling wrapping sklearn for speed with rigorous Qini curve evaluation.

UpliftML
Booking.com's enterprise uplift modeling via PySpark and H2O. Six meta-learners plus Uplift Random Forest with …

MaMiMo
Lightweight Python library focused specifically on Marketing Mix Modeling implementation.

Robyn
Meta's AI/ML-powered Marketing Mix Modeling package with ridge regression and multi-objective optimization

BTYDplus
Extended BTYD models for R including MBG/NBD, Pareto/GGG, and hierarchical Bayesian variants. Handles regular purchasing …
Geospatial Analysis
11 packages
OSMnx
Download, model, analyze, and visualize street networks and urban infrastructure from OpenStreetMap. Essential for …

conflictcartographer
Python package for conflict event data visualization and geospatial analysis
spacetrack
Python client for the Space-Track.org API to access satellite catalog and TLE data

(PySAL Core)
The broader PySAL ecosystem contains many tools for spatial data handling, weights, visualization, and analysis.

PySAL (spreg)
The spatial regression `spreg` module of PySAL. Implements spatial lag, error, IV models, and diagnostics.

Apache Sedona
Distributed spatial analytics engine (formerly GeoSpark) with spatial SQL, K-NN joins, and range queries for spatial …

Skyfield
Elegant astronomy library for computing satellite and celestial positions using JPL ephemeris data

sgp4
Implementation of the SGP4/SDP4 satellite propagation algorithms for processing TLE orbital data

spdep
The foundational R package for spatial weights matrix creation and spatial autocorrelation testing. Provides functions …
Transportation Modeling
9 packages
xlogit
GPU-accelerated estimation of mixed logit models using CuPy/NumPy. Orders of magnitude faster than traditional packages …

gtfs-kit
Analyze General Transit Feed Specification (GTFS) data. Compute route statistics, service frequencies, and visualize …
Apollo
Comprehensive R package for advanced choice modeling including mixed logit, latent class, hybrid choice, and integrated …

mlogit
The standard R package for multinomial logit estimation. Clean formula interface, nested logit support, and integration …
gmnl
R package for generalized multinomial logit models including G-MNL, LC-MNL, and MM-MNL for flexible preference …
mixl
Fast maximum simulated likelihood estimation of mixed logit models in R. Optimized for speed with large datasets.

tidytransit
Read and analyze GTFS transit feeds in the tidyverse style. Integrates with sf for spatial analysis and dplyr for data …
Sports Analytics
10 packages
hockeyR
R package for NHL play-by-play data with built-in expected goals models and player tracking statistics

worldfootballR
R package for scraping FBref, Transfermarkt, and Understat soccer data including xG, player values, and match statistics

nfl-data-py
Python package for accessing nflverse NFL play-by-play data with built-in EPA and win probability models

Lahman
R package providing the complete Lahman Baseball Database as native R data frames for seamless analysis

nflfastR
R package for NFL play-by-play data with built-in expected points (EPA) and win probability models from 1999-present

nba_api
Full NBA Stats API wrapper with 127+ endpoints for accessing shot charts, player tracking, play-by-play, and historical …

statsbombpy
Official Python API client for StatsBomb open data with 360 freeze-frame support for detailed soccer event analysis

mplsoccer
Python library for football/soccer pitch visualization with support for heat maps, shot maps, pass maps, and event …

pybaseball
Python library for pulling baseball data from Statcast, FanGraphs, Baseball Reference, and the Lahman database with …

hoopR
R package for accessing NBA Stats API plus ESPN and KenPom data for comprehensive basketball analytics
Healthcare Analytics
5 packages
BCEA
Bayesian Cost-Effectiveness Analysis in R. Processes MCMC output from JAGS/Stan, generates CEACs, CEAFs, and expected …

fhirclient
Official SMART on FHIR Python client. OAuth 2.0 authentication, resource CRUD operations, and search. Essential for …

MONAI
Medical Open Network for AI - PyTorch-based framework for deep learning in healthcare imaging. Domain-specific …

hesim
R package for health economic simulation modeling. Cohort discrete-time state transition models, partitioned survival …

heemod
Markov models for cost-effectiveness analysis in R. Define states, transitions, and costs/utilities with intuitive …
NLP & Language Models
16 packages
anthropic
Official Python SDK for Claude and Anthropic's API. Build AI applications with Claude models.

openai
Official Python SDK for OpenAI's API. Access GPT-4, o1, DALL-E, embeddings, and other OpenAI models.

crewai
Framework for orchestrating role-playing autonomous AI agents. Multi-agent collaboration made intuitive.

openai-agents
OpenAI's lightweight, production-ready SDK for building agentic AI applications. Fast prototyping.

langchain
Framework for developing LLM-powered applications. Chains, tools, memory, and retrieval.

spaCy
Industrial-strength NLP library for efficient text processing pipelines (NER, POS tagging, etc.).

langgraph
Framework for building stateful, multi-actor LLM applications. Graph-based agent workflows with persistence.

EDSL
Expected Parrot Domain-Specific Language for designing and running LLM-powered surveys and experiments. Create AI agent …

stm
Structural Topic Models incorporating document-level metadata as covariates affecting topic prevalence and content. …

tidytext
Tidy data principles for text mining. Converts text to tidy format (one-token-per-row), enabling analysis with dplyr, …
Gradient Boosting
13 packages
MLForecast
Scalable time series forecasting using machine learning models (e.g., LightGBM, XGBoost) as regressors.

xgboost
Extreme Gradient Boosting implementing state-of-the-art gradient boosted decision trees. Highly efficient, scalable, and …

XGBoost
High-performance, optimized gradient boosting library (also supports RF). Known for speed, efficiency, and winning …

xgboost
Extreme Gradient Boosting implementing state-of-the-art gradient boosted decision trees. Highly efficient, scalable, and …

NGBoost
Extends gradient boosting to probabilistic prediction, providing uncertainty estimates alongside point predictions. …

ranger
Fast implementation of random forests particularly suited for high-dimensional data. Provides survival forests, …

LightGBM
Fast, distributed gradient boosting (also supports RF). Known for speed, low memory usage, and handling large datasets.

LightGBM
Fast, distributed gradient boosting (also supports RF). Known for speed, low memory usage, and handling large datasets.

CatBoost
Gradient boosting library excelling with categorical features (minimal preprocessing needed). Robust against …
glmnet
Efficient procedures for fitting regularized generalized linear models via penalized maximum likelihood. Implements …
Recommender Systems
6 packages
DeepCTR
Easy-to-use implementations of deep CTR models including Wide&Deep, DeepFM, DIN, xDeepFM, and multi-task architectures

RecBole
Comprehensive recommendation library with 100+ algorithms spanning general, sequential, context-aware, and …

recommenderlab
R infrastructure for developing and evaluating recommender systems. Provides UBCF, IBCF, SVD, popular/random baselines …

Surprise
A Python scikit for building and analyzing recommender systems with explicit ratings. Implements SVD, SVD++, NMF, k-NN, …

Surprise
A Python scikit for building and analyzing recommender systems with explicit ratings. Implements SVD, SVD++, NMF, k-NN, …

LightFM
Hybrid recommendation library that handles cold-start by incorporating content features. Uses factorization machines to …

Implicit
GPU-accelerated library for collaborative filtering on implicit feedback data. Implements ALS, BPR, and logistic matrix …
Deep Learning
6 packages
DCA (Deep Count Autoencoder)
Denoising autoencoder for single-cell RNA-seq with ZINB output layer. Handles extreme sparsity in gene expression data.

TensorFlow
Google's end-to-end open-source machine learning platform. Build and deploy ML models at scale.

JAX
High-performance numerical computing with autograd and XLA compilation on CPU/GPU/TPU.

scVI-tools
Single-cell Variational Inference. Deep generative models for single-cell RNA-seq with ZINB likelihood handling 80-95% …

PyTorch
Popular deep learning framework with flexible automatic differentiation.

VaDE
Variational Deep Embedding. VAE with Gaussian Mixture Model prior in latent space for deep clustering.
Discrete Event Simulation
7 packages
SimPy
Process-based discrete-event simulation framework using Python generators. The standard for DES in Python with MIT …

Stable-Baselines3
Reliable PyTorch implementations of A2C, DDPG, DQN, PPO, SAC, TD3 RL algorithms. Published in JMLR 2021.

SuperSuit
Wrapper library for PettingZoo preprocessing including frame stacking, normalization, and action masking.

NetLogoR
Pure R implementation of NetLogo framework—no NetLogo installation required. Benefits from ggplot2 integration and R …

TorchRL
Official PyTorch reinforcement learning library with TensorDict abstraction for modular RL development.

Mesa
Leading open-source Python framework for agent-based modeling with spatial grids, agent schedulers, and Solara …

simmer
Process-oriented discrete-event simulation for R with C++ core via Rcpp. Supports magrittr pipe workflow for building …
Simulation Applications
14 packagesArena Simulation
Industry-leading discrete-event simulation software from Rockwell Automation. Used by majority of Fortune 100 companies …
RNetLogo
Embeds NetLogo into R for statistical analysis integration. Enables running NetLogo models and analyzing results in R …
AnyLogic
Multi-method simulation platform supporting discrete-event, agent-based, and system dynamics modeling. Free Personal …

CleanRL
Single-file RL algorithm implementations (~340 lines each) for educational purposes and research. Published in JMLR …

nlrx
rOpenSci package for NetLogo simulation via XML with BehaviorSpace support. Enables systematic NetLogo experiments from …

ABIDES
JPMorgan's agent-based interactive discrete event simulation for market microstructure research. NASDAQ-like exchange …

OR-Gym
Operations research environments for RL including knapsack, bin packing, supply chain, newsvendor, and portfolio …
Simio
Object-oriented discrete-event simulation with Process Digital Twin capabilities. Academic program offers free licenses …

ABCE
Agent-Based Computational Economics library from Oxford INET. Automatically handles trade with physically consistent …

pyNetLogo
Python-NetLogo interface enabling SALib sensitivity analysis integration and parallel NetLogo simulations. Published in …
Game Theory & Mechanisms
10 packages
OpenSpiel
DeepMind's 70+ game environments with multi-agent RL algorithms including Alpha-Rank, Neural Fictitious Self-Play, and …

pygambit
N-player extensive form games with Alan Turing Institute support. Computes Nash, perfect, and sequential equilibria.

Nashpy
Computation of Nash equilibria for 2-player games. Support enumeration and Lemke-Howson algorithm.

fairpy
Fair division algorithms from academic papers. Implements cake-cutting and item allocation procedures.

AuctionGym
Amazon's ad auction simulator for first/second-price auctions with RL bidding agents. Best Paper at AdKDD 2022.

deep-opt-auctions
Neural network optimal auction design. Implements RegretNet, RochetNet for mechanism design.

fairpyx
Course-seat allocation with capacity constraints. Practical fair division for university course assignment.

scarfmatch
Matching with couples using Scarf's algorithm. Essential for NRMP-style medical residency matching.

PettingZoo
Multi-agent version of Gymnasium with Agent-Environment-Cycle (AEC) model. Includes card games, MPE, and cooperative …

AI Economist
Salesforce's two-level RL environment for tax policy design. Published in Science Advances 2022. Includes COVID-19 …
Mathematical Optimization
6 packages
HiGHS
State-of-the-art open-source LP/MIP solver. Now the default solver in PyPSA, JuMP, and SciPy. Competitive with …

HiGHS
State-of-the-art open-source LP/MIP solver. Now the default solver in PyPSA, JuMP, and SciPy. Competitive with …

Pyomo
General-purpose algebraic optimization modeling in Python. Supports LP, MILP, NLP, and stochastic programming with …

Pyomo
General-purpose algebraic optimization modeling in Python. Supports LP, MILP, NLP, and stochastic programming with …

ortools
Google's operations research toolkit. Constraint programming, routing, linear/integer programming, and scheduling.
gurobipy
Python interface for Gurobi, the best-in-class commercial solver. LP, QP, MIP, and MIQP.

Faer
State-of-the-art linear algebra for Rust with Cholesky, QR, SVD decompositions and multithreaded solvers for large …

scipy.optimize
Optimization algorithms built into SciPy. Minimization, root finding, curve fitting, and linear programming.
Classical Inference
8 packageslmtest
Collection of tests for diagnostic checking in linear regression models. Provides the essential coeftest() function for …

Statsmodels
Comprehensive library for estimating statistical models (OLS, GLM, etc.), conducting tests, and data exploration. Core …
car
Functions accompanying 'An R Companion to Applied Regression.' Provides advanced regression diagnostics including …

lmerTest
Provides p-values for lme4 model fits via Satterthwaite's or Kenward-Roger degrees of freedom methods, with Type …

Linregress
Simple linear regression for Rust with R-style formula syntax, standard errors, t-stats, and p-values.

expectation
E-values and game-theoretic probability for sequential testing. Enables early signal detection with proper error …

broom
Converts messy output from 100+ statistical model types into consistent tidy tibbles using three verbs: tidy() for …

gtsummary
Creates publication-ready analytical and summary tables (Table 1 demographics, regression results, survival analyses) …
Modern Inference
11 packages
hypothetical
Library focused on hypothesis testing: ANOVA/MANOVA, t-tests, chi-square, Fisher's exact, nonparametric tests …

PyWhy-Stats
Part of the PyWhy ecosystem providing statistical methods specifically for causal applications, including various …

maketables
Publication-ready regression tables for pyfixest, statsmodels, linearmodels. Outputs HTML (great-tables), LaTeX, Word.

HypoRS
Hypothesis testing library for Rust with T-tests, Z-tests, ANOVA, Chi-square, designed to work seamlessly with Polars …

Statrs
Comprehensive statistical distributions for Rust (Normal, T, Gamma, etc.) with PDF, CDF, quantile functions—the …

Pingouin
User-friendly interface for common statistical tests (ANOVA, ANCOVA, t-tests, correlations, chi², reliability) built on …

Stargazer
Python port of R's stargazer for creating publication-quality regression tables (HTML, LaTeX) from `statsmodels` & …

marginaleffects
Modern standard for interpreting regression results—up to 1000× faster than margins. Computes marginal effects, …

texreg
Converts coefficients, standard errors, significance stars, and fit statistics from statistical models into LaTeX, HTML, …

Python Packages for Applied Economists
Curated collection of Python packages for applied researchers organized by functionality.
Resampling Methods
8 packagesboot
Classic bootstrap methods implementing the approaches described in Davison & Hinkley (1997). Provides functions for both …

fwildclusterboot
Fast wild cluster bootstrap implementation following Roodman et al. (2019)—up to 1000× faster than alternatives. …

rsample
Modern tidyverse-compatible resampling infrastructure. Provides functions for creating resamples (bootstrap, …

SciPy Bootstrap
(`scipy.stats.bootstrap`) Computes bootstrap confidence intervals for various statistics using percentile, BCa methods.

recombinator
Block bootstrap methods including Moving Block, Circular Block, Stationary, and Tapered Block Bootstrap for time series.
bootUR
Bootstrap unit root tests with sieve and wild bootstrap methods for time series stationarity testing.

wildboottest
Fast implementation of various wild cluster bootstrap algorithms (WCR, WCU) for robust inference, especially with few …
clusterbootstraps
Wild cluster bootstrap and pairs cluster bootstrap implementations for clustered standard errors.
Probabilistic Programming
12 packages
Bambi
High-level interface for building Bayesian GLMMs, built on top of PyMC. Uses formula syntax similar to R's `lme4`.
bsts
Bayesian Structural Time Series providing the foundation for CausalImpact. Supports spike-and-slab variable selection, …

NumPyro
Probabilistic programming library built on JAX for scalable Bayesian inference, often faster than PyMC.

PyMC
Flexible probabilistic programming library for Bayesian modeling and inference using MCMC algorithms (NUTS).

PyMC
Flexible probabilistic programming library for Bayesian modeling and inference using MCMC algorithms (NUTS).

rstanarm
Pre-compiled Bayesian regression models using Stan that mimic familiar R functions (lm, glm, lmer) with customary …

brms
High-level interface for fitting Bayesian generalized multilevel models using Stan, with lme4-style formula syntax …

bayesplot
Extensive library of ggplot2-based plotting functions for posterior analysis, MCMC diagnostics, and prior/posterior …

rstan
Core R interface to the Stan probabilistic programming language, providing full Bayesian inference via NUTS/HMC, …

Pyro
Deep universal probabilistic programming on PyTorch. Special support for Bayesian neural networks, normalizing flows, …
Survival Analysis
8 packages
mstate
Multi-state models in R. Handles competing risks, illness-death models, and complex disease progressions. Estimation, …

survminer
Visualization tools for survival analysis in R with publication-ready Kaplan-Meier plots, risk tables, and Cox model …

survHE
Survival analysis for health economics in R. Fits multiple parametric distributions, extrapolates survival curves, and …

scikit-survival
Machine learning for survival analysis compatible with scikit-learn, including gradient boosted models, random survival …

scikit-survival
Machine learning for survival analysis compatible with scikit-learn, including gradient boosted models, random survival …

survival
Core R package for survival analysis with Cox regression, Kaplan-Meier estimation, and parametric survival models - the …

survival
Core R package for survival analysis with Cox regression, Kaplan-Meier estimation, and parametric survival models - the …

flexsurv
Flexible parametric survival models including spline-based hazards, multi-state models, and cure models for complex …

pycox
PyTorch-based survival analysis. Implements DeepSurv, DeepHit, Cox-Time, and other neural survival models with partial …

lifelines
Comprehensive library for survival analysis: Kaplan-Meier, Nelson-Aalen, Cox regression, AFT models, handling censored …
Synthetic Data
10 packages
DataSynthesizer
Privacy-preserving synthetic data using Bayesian networks with differential privacy. From University of Washington …

Synthpop
Port of the R package for generating synthetic populations based on sample survey data.

CTGAN
GAN-based tabular data synthesizer using Variational GMM for mode-specific normalization. Published at NeurIPS 2019. …

Faker
Comprehensive fake data generator for 50+ locales including names, addresses, financial data, and more. Most popular …

DeepEcho
Time series synthetic data generation using deep learning. Part of the SDV ecosystem for sequential data.

Gretel Synthetics
Open-source synthetic data library with DGAN for time series, ACTGAN, and differential privacy support from Gretel.ai.

Mimesis
High-performance fake data generator—faster than Faker. Provides data for multiple domains and 35+ locales.
simPop
Synthetic population simulation for EU-SILC style survey data. Creates realistic household and individual-level …

SDV (Synthetic Data Vault)
Comprehensive library for generating synthetic tabular, relational, and time series data using various models.

sdcMicro
Statistical Disclosure Control for microdata used by World Bank and census agencies. Comprehensive anonymization …
Network Analysis
7 packages
NetworkCausalTree
Estimates both direct treatment effects and spillover effects under clustered network interference (Bargagli-Stoffi et …

ggraph
Grammar of graphics for network data built on ggplot2. Provides layouts, geometries, and faceting specifically designed …

igraph
Comprehensive network analysis library with efficient algorithms for network creation, manipulation, and analysis. …

igraph
Comprehensive network analysis library with efficient algorithms for network creation, manipulation, and analysis. …

tidygraph
Tidy data interface for network/graph data. Extends dplyr verbs to work with nodes and edges, enabling pipe-friendly …
sna
Social network analysis tools including network visualization, centrality measures, and statistical models for network …
Research Rabbit
Free tool for discovering academic papers through network visualization of paper connections and co-authorships.

python-louvain
Community detection in large networks using the Louvain algorithm, applicable to defense network analysis
Conformal Prediction
5 packages
fortuna
AWS library for uncertainty quantification in deep learning. Bayesian and conformal methods.

TorchCP
PyTorch-native conformal prediction for DNNs, GNNs, and LLMs with GPU acceleration.

puncc
IRT Lab's library for predictive uncertainty with conformal prediction. Supports various conformal methods.

MAPIE
Scikit-learn-contrib library for conformal prediction intervals. Provides model-agnostic uncertainty quantification for …

crepes
Lightweight library for conformal regressors and predictive systems. Simple API for calibrated prediction intervals.
Classical Time Series
15 packages
urca
Implements unit root and cointegration tests commonly used in applied econometric analysis. Includes Augmented …

pmdarima
ARIMA modeling with automatic parameter selection (auto-ARIMA), similar to R's `forecast::auto.arima`.

ARCH
Specialized library for modeling and forecasting conditional volatility using ARCH, GARCH, EGARCH, and related models.

Kats
Broad toolkit for time series analysis, including multivariate analysis, detection (outliers, change points, trends), …

vars
Comprehensive package for Vector Autoregression (VAR), Structural VAR (SVAR), and Structural Vector Error Correction …

KFAS
State space modeling framework for exponential family time series with computationally efficient Kalman filtering, …
dlm
Maximum likelihood and Bayesian analysis of Normal linear state space models (Dynamic Linear Models). Features …

forecast
The foundational R package for univariate time series forecasting. Provides methods for exponential smoothing via state …

Metran
Specialized package for estimating Dynamic Factor Models (DFM) using state-space methods and Kalman filtering.

StatsForecast
Fast, scalable implementations of popular statistical forecasting models (ETS, ARIMA, Theta, etc.) optimized for …
Seasonal Decomposition
4 packages
LocalProjections
Community implementations of Jordà (2005) Local Projections for estimating impulse responses without VAR assumptions.

fable
A tidyverse-native forecasting framework providing ETS, ARIMA, and other models for tidy time series (tsibble objects). …

Augurs
Time series forecasting and analysis for Rust with ETS, MSTL decomposition, seasonality detection, outlier detection, …
strucchange
Testing, monitoring, and dating structural changes in linear regression models. Implements the generalized fluctuation …
Forecasting Tools
5 packages
TS-Flint
Two Sigma's time-series library for Spark with optimized temporal joins, as-of joins, and distributed OLS for …

Prophet
Forecasting procedure for time series with strong seasonality and trend components, developed by Facebook.

prophet
Automatic forecasting procedure based on an additive decomposable model with non-linear trends, yearly/weekly/daily …

NeuralForecast
Deep learning models (N-BEATS, N-HiTS, Transformers, RNNs) for time series forecasting, built on PyTorch Lightning.

PyKalman
Implements Kalman filter, smoother, and EM algorithm for parameter estimation, including support for missing values and …
Structural Econometrics & Estimation
Comprehensive Rust econometrics library with OLS, IV, panel data estimators, fixed effects, DiD, and heteroskedasticity-robust standard errors (HC0-HC3).
Python package for simulation and estimation of Rust (1987) bus engine replacement model. Implements the nested fixed point (NFXP) algorithm for dynamic discrete choice. The reference implementation for learning structural estimation.
Python wrapper/interface to Dynare for DSGE model solving. Bridge between Python workflows and Dynare.
Solve nonlinear heterogeneous agent models (HANK) with perfect foresight. Efficient perturbation and projection methods.
DSGE modeling tools inspired by R's gEcon. Automatic first-order condition derivation with Dynare export.
DSGE model simulation, filtering, and Bayesian estimation. Handles occasionally binding constraints.
Fast upper envelope scan for discrete-continuous dynamic programming. JAX and numba implementations.
JAX-compatible DC-EGM algorithm for discrete-continuous dynamic programming (Iskhakov et al. 2017).
General equilibrium structural gravity modeling for trade policy analysis. Only Python package for Anderson-van Wincoop GE gravity.
R package for testing revealed preference axioms (GARP, WARP, SARP) on consumer choice data. Detects violations of utility maximization assumptions.
Framework for describing and solving economic models (DSGE, OLG, etc.) using a declarative YAML-based format.
Toolkit for solving, simulating, and estimating models with heterogeneous agents (e.g., consumption-saving).
Inverse optimization library that infers utility/cost functions from observed decisions. Learns the objective function that makes observed choices optimal given constraints.
Lightweight inverse optimization library for fitting models where the forward problem is a linear program. Recovers utility function weights from observed optimal decisions.
Core library for quantitative economics: dynamic programming, Markov chains, game theory, numerical methods.
Multivariate convex regression, stochastic frontier analysis, and data envelopment analysis (DEA). Implements Afriat inequalities for global concavity constraints in convex nonparametric least squares.
Simulation and estimation of finite-horizon dynamic discrete choice (DDC) models (e.g., labor/education choice).
Causal Inference (Synthetic Control)
Provides rigorous prediction intervals for synthetic control methods following Cattaneo et al. (2021, 2025). Supports staggered adoption designs with valid uncertainty quantification.
Implements the Augmented Synthetic Control Method, which uses an outcome model (ridge regression by default) to correct for bias when pre-treatment fit is imperfect. Uniquely supports staggered adoption across multiple treated units via multisynth() function.
Extends synthetic control method to micro-level data with many units. Implements permutation inference and handles high-dimensional settings where traditional SCM struggles.
Implements synthetic difference-in-differences, a hybrid method combining insights from both DiD and synthetic control that reweights and matches pre-treatment trends. Provides improved robustness properties compared to either method alone by combining their strengths.
The original synthetic control method implementation for comparative case studies. Constructs a weighted combination of comparison units to create a synthetic counterfactual for estimating effects of interventions on a single treated unit, as used in seminal studies of California tobacco program and German reunification.
Implements generalized synthetic control with interactive fixed effects, extending SCM to multiple treated units with variable treatment timing. Uses factor models to impute counterfactuals, handling unbalanced panels and complex treatment patterns with latent factor structures.
Implements penalized synthetic control method from Abadie & L'Hour (2021). Adds regularization to improve pre-treatment fit and reduce interpolation bias in sparse donor pools.
Brings synthetic control method into the tidyverse with cleaner syntax and built-in placebo inference. Provides pipe-friendly workflows for SCM estimation and visualization.
Automates placebo tests and multi-treated-unit ATT calculations for synthetic control. Provides utilities for generating in-space and in-time placebos with visualization.
Program Evaluation Methods (DiD, SC, RDD)
DiD with general treatment patterns. Handles effective treatment timing beyond simple staggered adoption.
Doubly robust estimation for group-time conditional average treatment effects. UCB for heterogeneous DiD.
Synthetic control method implementation compatible with R's Synth and augsynth packages.
Fast synthetic control estimators for panel data problems. Optimized ATT estimation with multiple SC algorithms.
Python port of Google's R package for estimating causal effects of interventions on time series using Bayesian structural time-series models.
Implements modern difference-in-differences methods for staggered adoption designs (e.g., Callaway & Sant'Anna).
Implementation of synthetic control methods for comparative case studies when panel data is available.
TensorFlow Probability port of Google's CausalImpact. Bayesian structural time-series for intervention effects.
Python adaptation of the R `did` package. Implements multi-period DiD with staggered treatment timing (Callaway & Sant’Anna).
Implements advanced synthetic control methods: forward DiD, cluster SC, factor models, and proximal SC. Designed for single-treated-unit settings.
Changes‑in‑Changes (CiC) estimator for distributional treatment effects (Athey & Imbens 2006).
Lee (2009) sample‑selection bounds for treatment effects; trims treated distribution to match selection rates.
Toolkit for sharp RDD analysis, including bandwidth calculation and estimation, integrating with pandas.
Comprehensive tools for Regression Discontinuity Designs (RDD), including optimal bandwidth selection, estimation, inference.
Double/Debiased Machine Learning (DML)
High-dimensional inference under hidden confounding. Doubly debiased Lasso for valid inference.
Microsoft's distributed ML library with native Double ML (DoubleMLEstimator) for heterogeneous treatment effects at scale.
Implements the double/debiased ML framework (Chernozhukov et al.) for estimating causal parameters (ATE, LATE, POM) with ML nuisances.
Microsoft toolkit for estimating heterogeneous treatment effects using DML, causal forests, meta-learners, and orthogonal ML methods.
Double‑post Lasso estimator for high‑dimensional treatment effects (Belloni‑Chernozhukov‑Hansen 2014).
Debiased‑Lasso detector of heterogeneous treatment effects in randomized experiments.
Adaptive Experimentation & Bandits
Meta's open-source platform for adaptive experimentation. Bayesian optimization, multi-objective optimization, and automated experiment design. Built on BoTorch for AI-assisted experimentation.
Sequential testing with always-valid inference. Supports continuous monitoring of A/B tests.
Lightweight microframework for Bayesian bandits (Thompson Sampling) with support for contextual/restless/delayed rewards.
Implements a wide range of contextual bandit algorithms (linear, tree-based, neural) and off-policy evaluation methods.
Production-ready, scikit-learn style library for contextual & stochastic bandits with parallelism and simulation tools.
Framework for **offline evaluation (OPE)** of bandit policies using logged data. Implements IPS, DR, DM estimators.
Library for advanced bandit problems: X-armed bandits (continuous/structured action spaces) and online optimization.
Comprehensive research framework for single/multi-player MAB algorithms (stochastic, adversarial, contextual).
Causal Inference (DiD)
Implements group-time average treatment effects (ATT(g,t)) for staggered DiD designs with multiple periods and variation in treatment timing. Provides flexible aggregation into event-study plots or overall treatment effect estimates, addressing the well-documented negative weighting issues with conventional TWFE under staggered adoption.
Implements the imputation-based DiD estimator that first estimates Y(0) counterfactuals from untreated observations using two-way fixed effects, then imputes treatment effects for treated units. Avoids negative weighting problems of conventional TWFE under heterogeneous treatment effects.
Implements locally efficient doubly robust DiD estimators that combine inverse probability weighting and outcome regression for improved statistical properties. Handles both panel data and repeated cross-sections in the canonical 2x2 DiD setting with covariates, providing robustness against model misspecification.
Performs Goodman-Bacon decomposition showing how two-way fixed effects (TWFE) estimates are weighted averages of all possible 2×2 DiD comparisons. Essential for diagnosing negative weights problems in staggered adoption designs.
Provides the efficient estimator for randomized staggered rollout designs, offering optimal weighting schemes for treatment effect estimation. Also implements Callaway & Sant'Anna and Sun & Abraham estimators with design-based Fisher inference for randomized experiments.
Comprehensive Difference-in-Differences library with sklearn-like API and statsmodels-style output. Implements Callaway-Sant'Anna, Sun-Abraham, Synthetic DiD, Triple Difference (DDD), Honest DiD sensitivity analysis, and extensive robustness diagnostics.
Constructs robust confidence intervals for DiD and event-study designs under violations of parallel trends. Allows researchers to conduct sensitivity analysis by relaxing the parallel trends assumption using smoothness or relative magnitude restrictions on pre-trend violations.
High-performance implementation of Callaway & Sant'Anna estimators optimized for large datasets with millions of observations. Reduces computation time from hours to seconds while supporting time-varying covariates and multiple events per unit.
Fixed Effects Counterfactual Estimators (v2.0+) incorporating gsynth functionality. Supports treatment switching on/off with carryover effects, matrix completion methods, and Rambachan & Roth sensitivity analysis for parallel trends violations.
Extended Two-Way Fixed Effects estimator implementing Wooldridge (2021, 2023). Saturated interaction effects to overcome bias in vanilla TWFE for difference-in-differences with staggered treatment adoption. Built on pyfixest and marginaleffects.
Time Series Econometrics
Two Sigma's time-series library for Spark with optimized temporal joins, as-of joins, and distributed OLS for high-frequency data.
Implements unit root and cointegration tests commonly used in applied econometric analysis. Includes Augmented Dickey-Fuller, Phillips-Perron, KPSS, Elliott-Rothenberg-Stock, and Zivot-Andrews tests, plus Johansen's cointegration procedure for multivariate series.
State space modeling framework for exponential family time series with computationally efficient Kalman filtering, smoothing, forecasting, and simulation. Supports observations from Gaussian, Poisson, binomial, negative binomial, and gamma distributions.
Maximum likelihood and Bayesian analysis of Normal linear state space models (Dynamic Linear Models). Features numerically stable SVD-based algorithms for Kalman filtering and smoothing, plus tools for MCMC-based Bayesian inference including forward filtering backward sampling (FFBS).
Provides an interface for fitting dynamic linear regression models with extended formula syntax. Supports convenient lag operators L(), differencing d(), trend(), season(), and harmonic components while preserving time series attributes.
Implements time series filters for extracting trend and cyclical components. Includes Hodrick-Prescott, Baxter-King, Christiano-Fitzgerald, Butterworth, and trigonometric regression filters commonly used in macroeconomics and business cycle analysis.
Testing, monitoring, and dating structural changes in linear regression models. Implements the generalized fluctuation test framework (CUSUM, MOSUM, recursive estimates) and F-test framework (Chow test, supF, aveF, expF) with breakpoint estimation and confidence intervals.
Implements nonlinear autoregressive time series models including threshold AR (TAR/SETAR), smooth transition AR (STAR, LSTAR), and multivariate extensions (TVAR, TVECM). Enables regime-switching dynamics analysis with parametric and non-parametric approaches.
Comprehensive package for Vector Autoregression (VAR), Structural VAR (SVAR), and Structural Vector Error Correction (SVEC) models. Provides estimation, lag selection, diagnostic testing, forecasting, Granger causality analysis, impulse response functions, and forecast error variance decomposition.
Specialized library for modeling and forecasting conditional volatility using ARCH, GARCH, EGARCH, and related models.
Broad toolkit for time series analysis, including multivariate analysis, detection (outliers, change points, trends), feature extraction.
Community implementations of Jordà (2005) Local Projections for estimating impulse responses without VAR assumptions.
Causal Inference
Meta's geo-experimental methodology combining Augmented Synthetic Control with power analysis
Discrete Choice Models
Discrete choice modeling with deep learning (e.g., TasteNet) for assortment and pricing optimization. Supports single-choice and basket models on large datasets. Published in JOSS.
Maximum likelihood estimation of parametric models, with strong support for complex discrete choice models.
Desktop application for computational revealed preference analysis. GUI written in Python with Rust backend for performance. Analyzes choice datasets for rationality and behavioral consistency.
Tools for estimating demand for differentiated products using the Berry-Levinsohn-Pakes (BLP) method.
Flexible implementation of conditional/multinomial logit models with utilities for data preparation.
Fast estimation of Multinomial Logit and Mixed Logit models, optimized for performance.
Inference algorithms for probabilistic choice models based on Luce's choice axiom. Supports Bradley-Terry (pairwise comparisons), Plackett-Luce (partial rankings), and Network Choice Models.
PyTorch framework for flexible estimation of complex discrete choice models, leveraging GPU acceleration.
Insurance & Actuarial
Actuarial science functions for R including loss distributions, credibility theory, ruin theory, and simulation of compound models
Comprehensive toolkit for extreme value analysis with diagnostic plots, return level estimation, and non-stationary models for climate-related risks
Open-source catastrophe modeling platform used by major reinsurers, supporting custom hazard/vulnerability models with standardized data formats
Python library for actuarial reserving implementing chain-ladder, Bornhuetter-Ferguson, Cape Cod, and stochastic methods for loss reserve estimation
Flexible parametric survival models including spline-based hazards, multi-state models, and cure models for complex time-to-event data
Open-source actuarial library with complete life insurance projection models including term, whole life, universal life, and variable annuities
Machine learning for survival analysis compatible with scikit-learn, including gradient boosted models, random survival forests, and Cox neural networks
Core R package for survival analysis with Cox regression, Kaplan-Meier estimation, and parametric survival models - the foundation for time-to-event analysis
Visualization tools for survival analysis in R with publication-ready Kaplan-Meier plots, risk tables, and Cox model forest plots
Comprehensive R package for claims reserving methods including Mack, Munich, and bootstrap chain-ladder with full uncertainty quantification
Microsoft toolkit for assessing and improving ML model fairness, critical for insurance pricing compliance and avoiding discriminatory outcomes
Model-agnostic explainability using Shapley values for any ML model, essential for actuarial model interpretability and regulatory compliance
Compound Poisson linear models for insurance claims with exact zero mass - handles the mixed discrete-continuous nature of claims data
Functions for extreme value distributions including GEV, GPD, and point process models essential for catastrophe modeling
R package for life insurance mathematics including life tables, annuities, and insurance present value calculations following actuarial notation
Python library for life actuarial calculations including commutation functions, life annuities, and insurance present values
Bayesian Econometrics
High-level interface for building Bayesian GLMMs, built on top of PyMC. Uses formula syntax similar to R's `lme4`.
Probabilistic programming library built on JAX for scalable Bayesian inference, often faster than PyMC.
Flexible probabilistic programming library for Bayesian modeling and inference using MCMC algorithms (NUTS).
Healthcare Economics & Health-Tech
Bayesian Cost-Effectiveness Analysis in R. Processes MCMC output from JAGS/Stan, generates CEACs, CEAFs, and expected value of information calculations.
Medical Open Network for AI - PyTorch-based framework for deep learning in healthcare imaging. Domain-specific transforms, pre-built architectures (UNet, SegResNet), and MONAI Label for annotation.
Markov models for cost-effectiveness analysis in R. Define states, transitions, and costs/utilities with intuitive syntax. Includes DSA, PSA, and scenario analysis.
Multi-state models in R. Handles competing risks, illness-death models, and complex disease progressions. Estimation, prediction, and visualization.
Official SMART on FHIR Python client. OAuth 2.0 authentication, resource CRUD operations, and search. Essential for building apps that connect to EHR systems.
R package for health economic simulation modeling. Cohort discrete-time state transition models, partitioned survival analysis, and probabilistic sensitivity analysis with parallelization.
Survival analysis for health economics in R. Fits multiple parametric distributions, extrapolates survival curves, and integrates with cost-effectiveness models.
MarTech & Customer Analytics
Extended BTYD models for R including MBG/NBD, Pareto/GGG, and hierarchical Bayesian variants. Handles regular purchasing patterns and incorporates purchase timing.
R package for probabilistic CLV modeling. Implements Pareto/NBD and BG/NBD with time-varying covariates, spending models, and customer-level predictions.
Comprehensive recommendation library with 100+ algorithms spanning general, sequential, context-aware, and knowledge-based approaches. Built on PyTorch with unified data loading and evaluation.
Industry-standard library for CLV modeling. Implements BG/NBD, Pareto/NBD for transaction prediction and Gamma-Gamma for monetary value modeling in non-contractual settings.
R infrastructure for developing and evaluating recommender systems. Provides UBCF, IBCF, SVD, popular/random baselines with unified evaluation framework.
GPU-accelerated library for collaborative filtering on implicit feedback data. Implements ALS, BPR, and logistic matrix factorization with CUDA support for scale.
Hybrid recommendation library that handles cold-start by incorporating content features. Uses factorization machines to learn embeddings for users, items, and their features simultaneously.
Bunching Estimation
Implements Kleven-Waseem style bunching estimation for kink and notch designs. Calculates parametric elasticities from bunching at tax thresholds with publication-ready output.
Optimization
Numerical optimization framework for Rust with Newton, BFGS, L-BFGS, trust region, and derivative-free methods for MLE/GMM.
Python interface for Gurobi, the best-in-class commercial solver. LP, QP, MIP, and MIQP.
Google's operations research toolkit. Constraint programming, routing, linear/integer programming, and scheduling.
Domain-specific language for convex optimization problems. Write math as code — the standard for convex problems.
Optimization algorithms built into SciPy. Minimization, root finding, curve fitting, and linear programming.
Transportation Economics & Technology
Comprehensive R package for advanced choice modeling including mixed logit, latent class, hybrid choice, and integrated choice-latent variable models.
Download, model, analyze, and visualize street networks and urban infrastructure from OpenStreetMap. Essential for transportation network analysis.
Simulation of Urban Mobility - open source traffic simulation suite for modeling road networks, public transit, pedestrians, and multimodal scenarios.
Fast maximum simulated likelihood estimation of mixed logit models in R. Optimized for speed with large datasets.
The standard R package for multinomial logit estimation. Clean formula interface, nested logit support, and integration with R's modeling ecosystem.
Open source multimodal trip planning engine. Combines GTFS transit, OpenStreetMap streets, and bike-share for routing and isochrone analysis.
R package for generalized multinomial logit models including G-MNL, LC-MNL, and MM-MNL for flexible preference heterogeneity.
Analyze General Transit Feed Specification (GTFS) data. Compute route statistics, service frequencies, and visualize transit networks.
Read and analyze GTFS transit feeds in the tidyverse style. Integrates with sf for spatial analysis and dplyr for data manipulation.
GPU-accelerated estimation of mixed logit models using CuPy/NumPy. Orders of magnitude faster than traditional packages for large datasets.
Simulation & Computational Economics
Agent-Based Computational Economics library from Oxford INET. Automatically handles trade with physically consistent goods, includes built-in Firm/Household archetypes.
JPMorgan's agent-based interactive discrete event simulation for market microstructure research. NASDAQ-like exchange with multiple agent types.
Salesforce's two-level RL environment for tax policy design. Published in Science Advances 2022. Includes COVID-19 economic simulation.
Amazon's ad auction simulator for first/second-price auctions with RL bidding agents. Best Paper at AdKDD 2022.
Farama Foundation's successor to OpenAI Gym. Standard single-agent reinforcement learning API for environment development and benchmarking.
Multi-objective reinforcement learning environments for Pareto-optimal policy learning with conflicting objectives.
Leading open-source Python framework for agent-based modeling with spatial grids, agent schedulers, and Solara visualization. Mesa 3 (2025) requires Python 3.12+.
Operations research environments for RL including knapsack, bin packing, supply chain, newsvendor, and portfolio optimization.
Multi-agent version of Gymnasium with Agent-Environment-Cycle (AEC) model. Includes card games, MPE, and cooperative environments. NeurIPS 2021.
Industry-grade scalable reinforcement learning library from Ray. Native multi-agent support for distributed training at scale.
Reliable PyTorch implementations of A2C, DDPG, DQN, PPO, SAC, TD3 RL algorithms. Published in JMLR 2021.
Wrapper library for PettingZoo preprocessing including frame stacking, normalization, and action masking.
Model-based trading environments for market-making and optimal execution RL. Implements Avellaneda-Stoikov and Cartea-Jaimungal models.
rOpenSci package for NetLogo simulation via XML with BehaviorSpace support. Enables systematic NetLogo experiments from R.
Python-NetLogo interface enabling SALib sensitivity analysis integration and parallel NetLogo simulations. Published in JASSS (2018).
Modern Python framework for agent-based modeling integrating model design with SALib sensitivity analysis and NetworkX network structures.
Multi-method simulation platform supporting discrete-event, agent-based, and system dynamics modeling. Free Personal Learning Edition available.
Industry-leading discrete-event simulation software from Rockwell Automation. Used by majority of Fortune 100 companies for process optimization.
Discrete-event simulation library specializing in open queueing networks. Supports multiple customer classes, blocking, baulking, reneging, and priority classes.
Single-file RL algorithm implementations (~340 lines each) for educational purposes and research. Published in JMLR 2022.
First open-source deep reinforcement learning framework for quantitative finance. Train-test-trade pipeline for NASDAQ, DJIA, S&P 500.
Pure R implementation of NetLogo framework—no NetLogo installation required. Benefits from ggplot2 integration and R spatial objects.
Embeds NetLogo into R for statistical analysis integration. Enables running NetLogo models and analyzing results in R environment.
Process-based discrete-event simulation framework using Python generators. The standard for DES in Python with MIT license, requiring Python 3.8+.
Object-oriented discrete-event simulation with Process Digital Twin capabilities. Academic program offers free licenses for teaching.
Official PyTorch reinforcement learning library with TensorDict abstraction for modular RL development.
Analytical solver for Markovian queueing models and product-form queueing networks in R. Computes steady-state probabilities and performance metrics.
Process-oriented discrete-event simulation for R with C++ core via Rcpp. Supports magrittr pipe workflow for building simulation models fluently.
Numerical Optimization & Computational Tools
JAX-ecosystem implementations of standard econometrics routines for GPU computation.
Econometrics implementations in PyTorch. Leverages autodiff and GPU acceleration for econometric methods.
State-of-the-art linear algebra for Rust with Cholesky, QR, SVD decompositions and multithreaded solvers for large systems.
High-performance numerical computing with autograd and XLA compilation on CPU/GPU/TPU.
General-purpose linear algebra library for Rust with dense and sparse matrices, widely used in graphics and physics.
N-dimensional array library for Rust—the NumPy equivalent with slicing, broadcasting, and BLAS/LAPACK integration.
Popular deep learning framework with flexible automatic differentiation.
Experimental Design
Proper randomization procedures for experiments with known assignment probabilities. Implements simple, complete, block, and cluster randomization with exact probability calculations for IPW estimation.
Ex ante experimental design declaration and diagnosis. Enables researchers to formally describe their research design, diagnose statistical properties via simulation, and improve designs before data collection.
Multi-armed bandit algorithms including Thompson Sampling, UCB, and LinUCB. Directly applicable to adaptive A/B testing and recommendation optimization with simulation and evaluation tools.
Simulates realistic social science data for power analysis and design testing. Creates hierarchical data structures with correlated variables matching real-world patterns.
Energy & Utilities Economics
Python wrapper for the EIA Open Data API. Access generation, consumption, prices, and other energy data programmatically.
State-of-the-art open-source LP/MIP solver. Now the default solver in PyPSA, JuMP, and SciPy. Competitive with commercial solvers on many problem types.
EPRI's open-source distribution system simulator. Quasi-static time-series analysis, DER integration, and comprehensive distribution modeling. Industry standard.
General-purpose algebraic optimization modeling in Python. Supports LP, MILP, NLP, and stochastic programming with interfaces to major solvers including HiGHS, Gurobi, and CPLEX.
Capacity expansion model from MIT/Princeton in Julia. Highly configurable with unit commitment, long-duration storage, and transmission expansion. Used for Net-Zero America.
Power network optimization in Julia. Supports AC/DC optimal power flow, transmission expansion, and custom formulations with strong mathematical rigor.
Python for Power System Analysis - the workhorse for large-scale power system optimization. Static power flow, linear OPF, capacity expansion, unit commitment, and storage modeling.
Public Utility Data Liberation - cleaned, integrated U.S. energy data. Combines EIA, FERC, and EPA data into analysis-ready formats with comprehensive documentation.
Power system analysis for distribution networks. Newton-Raphson power flow, state estimation, short circuit calculations, and network visualization.
Data Access
Unified Python interface for U.S. electricity grid data from all major ISOs
Geo-Experiments & Lift Measurement
Google's time-based regression with greedy search for optimal geo experiment groups.
Google's robust analysis for paired geo experiments using trimmed statistics. Handles outliers in geo-level data.
Causal Discovery & Graphical Models
Benchmarking 41+ structure learning algorithms for causal discovery. Standardized evaluation framework.
Mixture of Causal Graphs discovery for heterogeneous time series (ICML 2024). Finds time-varying causal structures.
State-dependent causal inference for conditionally stationary processes (ICML 2025). Handles regime-switching causal graphs.
Bayesian network structure learning, parameter estimation, and inference. Implements constraint-based (PC, GS), score-based (HC, TABU), and hybrid algorithms for DAG learning with discrete and continuous data.
LLM + BFS hybrid for efficient causal graph discovery. Uses language models to guide structure search.
Analysis of structural causal models represented as DAGs. Computes adjustment sets, identifies instrumental variables, tests conditional independencies, and finds minimal sufficient adjustment sets for causal identification.
Visualize and analyze causal DAGs using ggplot2. Provides tidy interface to dagitty with publication-quality DAG plots, path highlighting, and adjustment set visualization.
Causal structure learning from observational data using the PC algorithm and variants. Estimates Markov equivalence class of DAGs from conditional independence tests with intervention support.
Causal inference using graphical models (DAGs), including identification theory and effect estimation.
Implements algorithms for causal discovery (recovering causal graph structure) from observational data.
Uses Bayesian Networks for causal reasoning, combining ML with expert knowledge to model relationships.
Specialized package for learning non-Gaussian linear causal models, implementing various versions of the LiNGAM algorithm including ICA-based methods.
Specialized package for causal inference in time series data implementing PCMCI, PCMCIplus, LPCMCI algorithms with conditional independence tests.
Comprehensive Python package serving as Python translation and extension of Java-based Tetrad toolkit for causal discovery algorithms.
Huawei Noah's Ark Lab end-to-end causal structure learning toolchain emphasizing gradient-based methods with GPU acceleration (NOTEARS, GOLEM).
Python interface to Tetrad Java library using JPype, providing direct access to Tetrad's causal discovery algorithms with efficient data translation.
Agentic AI
Official Python SDK for Claude and Anthropic's API. Build AI applications with Claude models.
Official Python SDK for OpenAI's API. Access GPT-4, o1, DALL-E, embeddings, and other OpenAI models.
Framework for developing LLM-powered applications. Chains, tools, memory, and retrieval.
Expected Parrot Domain-Specific Language for designing and running LLM-powered surveys and experiments. Create AI agent personas with demographic traits for homo silicus research.
Framework for orchestrating role-playing autonomous AI agents. Multi-agent collaboration made intuitive.
Framework for building stateful, multi-actor LLM applications. Graph-based agent workflows with persistence.
OpenAI's lightweight, production-ready SDK for building agentic AI applications. Fast prototyping.
Natural Language Processing for Economics
Causal inference for text data. Estimate treatment effects from unstructured text using NLP.
Framework for state-of-the-art sentence, text and image embeddings. Powers semantic search and similarity applications.
Library focused on topic modeling (LDA, LSI) and document similarity analysis.
Natural Language Toolkit - comprehensive library for NLP research and education with 50+ corpora and lexical resources.
Access to thousands of pre-trained models for NLP tasks like text classification, summarization, embeddings, etc.
Industrial-strength NLP library for efficient text processing pipelines (NER, POS tagging, etc.).
Meta-Analysis & Systematic Review
General package for meta-analysis providing fixed and random effects models, subgroup analysis, meta-regression, and publication bias tests. User-friendly interface for common meta-analytic tasks.
Automated search term identification for systematic reviews via keyword co-occurrence networks. Helps build comprehensive search strategies by identifying relevant terms from seed articles.
The gold standard for meta-analysis in R. Fixed/random/mixed-effects models, multivariate meta-analysis, network meta-analysis, forest/funnel/GOSH plots. Comprehensive and well-documented.
Comprehensive systematic review toolkit with multi-reviewer assignment, figure extraction from PDFs, PRISMA diagram generation, and effect size calculation. Designed for research synthesis workflows.
Tools for evidence synthesis and systematic reviews. Duplicate detection, topic modeling for screening, title/abstract screening GUI, and import from multiple bibliography formats.
Causal Inference & Matching
Doubly robust, non-parametric estimation of node-wise counterfactual means under network interference (arXiv 2024).
Estimates both direct treatment effects and spillover effects under clustered network interference (Bargagli-Stoffi et al. 2025).
Implements Cinelli-Hazlett framework for assessing robustness to unobserved confounding. Computes confounder strength needed to invalidate results.
Minimal, fast AIPW (Augmented Inverse Probability Weighting) implementation for discrete treatments. Sklearn-compatible with cross-fitting.
Continuous treatment dose-response curve estimation. GPS and TMLE methods for continuous treatments.
Comprehensive Python implementation for heterogeneous treatment effect estimation. Handles binary/multiple discrete treatments with optimal policy learning via Policy Trees.
Dynamic treatment regimes using Iterative Q-Learning. Scikit-learn compatible for multi-stage optimal treatment sequencing.
Regression and ML adjustments to treatment effects in RCTs. Implements List et al. (2024) methods.
JAX-accelerated neural network CATE estimators implementing SNet, FlexTENet, TARNet, CFRNet, and DragonNet architectures.
Implements classical causal inference methods like propensity score matching, inverse probability weighting, stratification.
IBM-developed package that provides a scikit-learn-inspired API for causal inference with meta-algorithms supporting arbitrary machine learning models.
Focuses on uplift modeling and heterogeneous treatment effect estimation using machine learning techniques.
Implements Propensity Score Matching (PSM) and Coarsened Exact Matching (CEM) with ML flexibility for propensity score estimation.
Python library for causal research that addresses the scarcity of real-world datasets with known causal relations. Provides fine-grained control over structural causal models.
Developed by PyMC Labs, focuses specifically on causal inference in quasi-experimental settings. Specializes in scenarios where randomization is impossible or expensive.
End-to-end framework for causal inference based on causal graphs (DAGs) and potential outcomes. Covers identification, estimation, refutation.
Fast k-nearest-neighbor matching for large datasets using Facebook's FAISS library.
Focuses on uplift modeling and estimating heterogeneous treatment effects using various ML-based methods.
Causal inference framework providing tools for causal graph manipulation and effect identification.
Panel Data & Fixed Effects
Panel data modeling with IV tests (weak IV, over-identification, endogeneity) and 2-step GMM estimation.
Fast and comprehensive package for estimating econometric models with multiple high-dimensional fixed effects, including OLS, GLM, Poisson, and negative binomial models. Features native support for clustered standard errors (up to four-way), instrumental variables, and modern difference-in-differences estimators including Sun-Abraham for staggered treatments.
Heterogeneity analysis across units in panel data. Detects and characterizes unit-level variation.
Automates within-between (hybrid) model specification for panel/longitudinal data, combining fixed effects robustness to time-invariant confounding with random effects ability to estimate time-invariant coefficients. Uses lme4 for multilevel estimation with optional Bayesian (brms) and GEE (geepack) backends.
Solves linear models with high-dimensional fixed effects, supporting robust variance calculation and IV.
Estimation of fixed, random, pooled OLS models for panel data. Also Fama-MacBeth and between/first-difference estimators.
Fast estimation of linear models with multiple high-dimensional fixed effects (like R's `fixest`). Supports OLS, IV, Poisson, robust/cluster SEs.
Fits generalized linear models (Poisson, negative binomial, logit, probit, Gamma) with high-dimensional k-way fixed effects. Partials out factors during log-likelihood optimization and provides robust/multi-way clustered standard errors, fixed effects recovery, and analytical bias corrections for binary choice models.
Estimates fixed effects binary choice models (logit and probit) with potentially many individual fixed effects using a pseudo-demeaning algorithm. Addresses the incidental parameters problem through analytical bias correction based on Fernández-Val (2009) and computes average partial effects.
Out-of-core regression (OLS/IV) for very large datasets using DuckDB aggregation. Handles data that doesn't fit in memory.
Efficiently estimates linear models with multiple high-dimensional fixed effects using the Method of Alternating Projections. Designed for datasets with factors having thousands of levels (hundreds of thousands of dummy variables), with full support for 2SLS instrumental variables and multi-way clustered standard errors.
Comprehensive econometrics package for linear panel models providing fixed effects (within), random effects, between, first-difference, Hausman-Taylor, and nested random effects estimators. Includes GMM, FGLS, and extensive diagnostic tests for serial correlation, cross-sectional dependence, and panel unit roots.
Estimation of dynamic panel data models using Arellano-Bond (Difference GMM) and Blundell-Bond (System GMM). Includes Windmeijer correction & tests.
Causal Inference (Matching)
Implements Covariate Balancing Propensity Score, which estimates propensity scores by jointly optimizing treatment prediction and covariate balance via generalized method of moments (GMM). Supports binary, multi-valued, and continuous treatments, as well as longitudinal settings for marginal structural models.
Comprehensive matching package that selects matched samples of treated and control groups with similar covariate distributions. Provides a unified interface to multiple matching methods including nearest neighbor, optimal pair, optimal full, genetic, exact, coarsened exact (CEM), cardinality matching, and subclassification with propensity score estimation via GLM, GAM, random forest, and BART.
Generates standardized balance tables and plots for covariates after preprocessing via matching, weighting, or subclassification. Provides unified balance assessment across multiple R packages (MatchIt, WeightIt, twang, Matching, optmatch, CBPS, ebal, cem, sbw, designmatch). Supports multi-category, continuous, and longitudinal treatments with clustered and multiply imputed data.
Implements entropy balancing, a reweighting method that finds weights for control units such that specified covariate moment conditions (means, variances) are exactly satisfied while staying as close as possible to uniform weights by minimizing Kullback-Leibler divergence. Primarily designed for ATT estimation.
Distance-based bipartite matching using minimum cost network flow algorithms, oriented to matching treatment and control groups in observational studies. Provides optimal full matching and pair matching with support for propensity score distances, Mahalanobis distance, calipers, and exact matching constraints.
Unified interface for generating balancing weights for causal effect estimation in observational studies. Supports binary, multi-category, and continuous treatments for point and longitudinal/marginal structural models. Methods include inverse probability weighting (IPW), entropy balancing, covariate balancing propensity score (CBPS), energy balancing, stable balancing weights, BART, and SuperLearner.
Causal Inference (RDD)
Provides tools for power, sample size, and minimum detectable effects (MDE) calculations in RD designs using robust bias-corrected local polynomial inference: rdpower() calculates power, rdsampsi() calculates required sample size for desired power, and rdmde() computes minimum detectable effects.
Implements manipulation testing (density discontinuity testing) procedures using local polynomial density estimators to detect perfect self-selection around a cutoff. Provides rddensity() for hypothesis testing, rdbwdensity() for bandwidth selection, and rdplotdensity() for density plots with confidence bands.
Regression discontinuity design toolkit with clustered inference for geographic discontinuities. Provides bandwidth selection, specification tests, and visualization tools.
Provides tools for RD analysis under local randomization: rdrandinf() performs hypothesis testing using randomization inference, rdwinselect() selects a window around the cutoff where randomization likely holds, rdsensitivity() assesses sensitivity to different windows, and rdrbounds() constructs Rosenbaum bounds for unobserved confounders.
Provides tools for RD designs with multiple cutoffs or scores: rdmc() estimates pooled and cutoff-specific effects in multi-cutoff designs, rdmcplot() draws RD plots for multi-cutoff designs, and rdms() estimates effects in cumulative cutoffs or multi-score (geographic/boundary) designs.
Supports multi-assignment RDD with two running variables, power analysis for RDD designs, and includes a Shiny interface for interactive analysis. Handles both sharp and fuzzy designs with bandwidth selection.
# Search code, repositories, users, issues, pull requests... You signed in with another tab or window. You signed out in another tab or window. You switched accounts on another tab or window. # rdpackages/rdhte. | Name | Name | Last commit message | Last commit date |. | Latest commit History23 Co
Conformal Prediction & Uncertainty
Scikit-learn-contrib library for conformal prediction intervals. Provides model-agnostic uncertainty quantification for regression and classification.
PyTorch-native conformal prediction for DNNs, GNNs, and LLMs with GPU acceleration.
AWS library for uncertainty quantification in deep learning. Bayesian and conformal methods.
IRT Lab's library for predictive uncertainty with conformal prediction. Supports various conformal methods.
Lightweight library for conformal regressors and predictive systems. Simple API for calibrated prediction intervals.
Instrumental Variables
Implements correct standard errors for Bartik/shift-share instrumental variables designs following Adão, Kolesár, and Morales (2019 QJE). Standard clustered SEs are typically incorrect for shift-share—this package provides econometrically valid inference.
Generalized Method of Moments estimation implementing two-step GMM, iterated GMM, and continuous updated estimator (CUE) with HAC covariance matrices. Supports linear and nonlinear moment conditions.
Specialized package for weak instrument diagnostics implementing Anderson-Rubin tests, k-class estimators (LIML, Fuller), and sensitivity analysis following Jiang et al. (2015). Essential when instrument strength is questionable.
Modern implementation of two-stage least squares (2SLS) instrumental variables regression with comprehensive diagnostics including hat values, studentized residuals, and component-plus-residual plots. Successor to AER's ivreg() function with superior diagnostic tools.
Modern S4-based implementation of Generalized Method of Moments supporting systems of equations, nonlinear moment conditions, and hypothesis testing. Successor to gmm package with object-oriented design.
Simultaneous systems estimation implementing Seemingly Unrelated Regression (SUR), two-stage least squares (2SLS), and three-stage least squares (3SLS). Critical for demand systems and structural macro models.
Bayesian Inference
Extensive library of ggplot2-based plotting functions for posterior analysis, MCMC diagnostics, and prior/posterior predictive checks supporting the applied Bayesian workflow for any MCMC-fitted model.
High-level interface for fitting Bayesian generalized multilevel models using Stan, with lme4-style formula syntax supporting linear, count, survival, ordinal, zero-inflated, hurdle, and mixture models with flexible prior specification.
Core R interface to the Stan probabilistic programming language, providing full Bayesian inference via NUTS/HMC, approximate inference via ADVI, and penalized maximum likelihood via L-BFGS for custom Bayesian models.
Pre-compiled Bayesian regression models using Stan that mimic familiar R functions (lm, glm, lmer) with customary formula syntax, weakly informative default priors, and zero model compilation time.
Deep universal probabilistic programming on PyTorch. Special support for Bayesian neural networks, normalizing flows, and stochastic variational inference.
Bootstrap & Inference
Classic bootstrap methods implementing the approaches described in Davison & Hinkley (1997). Provides functions for both parametric and nonparametric resampling with various confidence interval methods.
Bootstrap unit root tests with sieve and wild bootstrap methods for time series stationarity testing.
Fast wild cluster bootstrap implementation following Roodman et al. (2019)—up to 1000× faster than alternatives. Critical for panel data with few clusters. Integrates with fixest and lfe for efficient inference.
Block bootstrap methods including Moving Block, Circular Block, Stationary, and Tapered Block Bootstrap for time series.
Modern tidyverse-compatible resampling infrastructure. Provides functions for creating resamples (bootstrap, cross-validation, time series splits) that integrate seamlessly with tidymodels workflows.
Causal Inference (ML)
Conducts sensitivity analyses for unmeasured confounding, selection bias, and measurement error in observational studies and meta-analyses. Computes E-values representing the minimum strength of association unmeasured confounders would need to fully explain away an observed effect.
Implements the Super Learner algorithm for optimal ensemble prediction via cross-validation. Creates weighted combinations of multiple ML algorithms (XGBoost, Random Forest, glmnet, neural networks, SVM, BART) with guaranteed asymptotic optimality.
Semiparametric causal inference methods based on inverse probability weighting and double machine learning for average treatment effects, causal mediation analysis (direct/indirect effects), and dynamic treatment evaluation. Supports LATE estimation with instrumental variables.
Streamlined double/debiased machine learning estimation with emphasis on (short-)stacking to combine multiple base learners, increasing robustness to unknown data generating processes. Designed as a complement to DoubleML with simpler syntax.
High-dimensional statistical methods featuring heteroscedasticity-robust LASSO with theoretically-grounded penalty selection, post-double-selection inference, and treatment effect estimation under sparsity assumptions for high-dimensional controls.
Targeted maximum likelihood estimation for treatment/censoring-specific mean outcomes with time-varying treatments and confounders. Supports longitudinal settings, marginal structural models, and dynamic treatment regimes alongside IPTW and G-computation.
Suite of sensitivity analysis tools extending the traditional omitted variable bias framework, computing robustness values, bias-adjusted estimates, and sensitivity contour plots for OLS regression to assess how strong unmeasured confounders would need to be to overturn conclusions.
Implements targeted maximum likelihood estimation for point treatment effects with binary or continuous outcomes. Estimates ATE, ATT, ATC, and supports marginal structural models. Integrates SuperLearner for data-adaptive nuisance parameter estimation.
A modular, extensible framework for targeted minimum loss-based estimation supporting custom TMLE parameters through a unified interface. Part of the tlverse ecosystem, designed to be as general as the mathematical TMLE framework itself for complex analyses.
BERT-based causal inference from text. Implements methods from Veitch et al. (2020) showing representations must predict both treatment and outcome.
Implements meta-learner algorithms (S-learner, T-learner, X-learner) for heterogeneous treatment effect estimation using flexible base learners including honest Random Forests and BART for personalized CATE estimation.
Causal Forests with Fixed Effects (CFFE) for estimating heterogeneous treatment effects in panel data and difference-in-differences settings. Node-level fixed-effect residualization during tree construction.
Forest-based statistical estimation and inference for heterogeneous treatment effects, supporting multiple treatment arms, instrumental variables, survival outcomes, and quantile regression—all with honest estimation and valid confidence intervals. The most widely-used R package for CATE estimation.
Production-grade Python implementation of Generalized Random Forests for heterogeneous treatment effects. Multiple performance tiers (pure Python, Numba, Cython) with valid statistical inference via Infinitesimal Jackknife.
Power Simulation & Design of Experiments
End-to-end A/B testing from MobileTeleSystems with PySpark support. Covers experiment design, multi-group splitting, matching, and inference.
Simulation-based power analysis for factorial ANOVA designs (up to 3 factors). Includes Shiny app for interactive power analysis.
Machine learning-based power analysis using surrogate models. Efficient sample size planning for complex study designs.
DGP (Data Generating Process) framework for systematic simulation studies. Enables reproducible computational experiments.
Calculate A/B test statistics directly within data warehouses (BigQuery, ClickHouse, Snowflake, Spark) via Ibis interface. Supports CUPED/CUPAC.
Bayesian Adaptive Design Optimization (ADO) for tuning experiments in real-time, with models for psychometric tasks.
Parallel active learning library for adaptive function sampling/evaluation, with live plotting for monitoring.
Automates generation and optimization of designs, especially for mixed factor-level experiments; computes efficiency metrics.
Implements classical Design of Experiments: factorial (full/fractional), response surface (Box-Behnken, CCD), Latin Hypercube.
Statistical Inference & Hypothesis Testing
Hypothesis testing library for Rust with T-tests, Z-tests, ANOVA, Chi-square, designed to work seamlessly with Polars DataFrames.
Gaussian copula imputation for mixed variable types with streaming capability (Journal of Statistical Software 2024).
LightGBM-accelerated multiple imputation by chained equations. Fast MICE for large datasets.
Safe Anytime Valid Inference using e-processes and confidence sequences (Ramdas et al. 2023). Valid inference at any stopping time.
User-friendly interface for common statistical tests (ANOVA, ANCOVA, t-tests, correlations, chi², reliability) built on pandas & scipy.
Part of the PyWhy ecosystem providing statistical methods specifically for causal applications, including various independence tests and power-divergence methods.
Foundational module within SciPy for a wide range of statistical functions, distributions, and hypothesis tests (t-tests, ANOVA, chi², KS, etc.).
Comprehensive statistical distributions for Rust (Normal, T, Gamma, etc.) with PDF, CDF, quantile functions—the scipy.stats equivalent.
E-values and game-theoretic probability for sequential testing. Enables early signal detection with proper error control.
Library focused on hypothesis testing: ANOVA/MANOVA, t-tests, chi-square, Fisher's exact, nonparametric tests (Mann-Whitney, Kruskal-Wallis, etc.).
Comprehensive library for survival analysis: Kaplan-Meier, Nelson-Aalen, Cox regression, AFT models, handling censored data.
State Space & Volatility Models
Focuses on Kalman filters (standard, EKF, UKF) and smoothers with a clear, pedagogical implementation style.
Specialized package for estimating Dynamic Factor Models (DFM) using state-space methods and Kalman filtering.
Implements Kalman filter, smoother, and EM algorithm for parameter estimation, including support for missing values and UKF.
(See Bayesian) Bayesian state-space modeling using PyMC, integrating Kalman filtering within MCMC for parameter estimation.
Efficient Bayesian estimation of stochastic volatility (SV) models using MCMC.
Machine Learning
Denoising autoencoder for single-cell RNA-seq with ZINB output layer. Handles extreme sparsity in gene expression data.
Variational Deep Embedding. VAE with Gaussian Mixture Model prior in latent space for deep clustering.
Fast implementation of random forests particularly suited for high-dimensional data. Provides survival forests, classification, and regression with efficient memory usage. Core backend for grf's causal forests.
Modern framework for modeling and machine learning using tidyverse principles. Meta-package including parsnip (model specification), recipes (preprocessing), workflows, tune (hyperparameter tuning), and yardstick (metrics). Successor to caret.
Extreme Gradient Boosting implementing state-of-the-art gradient boosted decision trees. Highly efficient, scalable, and portable with interfaces to R, Python, and other languages. Essential for prediction in double ML workflows.
Google's end-to-end open-source machine learning platform. Build and deploy ML models at scale.
Efficient procedures for fitting regularized generalized linear models via penalized maximum likelihood. Implements LASSO, ridge regression, and elastic net with extremely fast coordinate descent algorithms. Foundation for high-dimensional regression and causal ML.
Single-cell Variational Inference. Deep generative models for single-cell RNA-seq with ZINB likelihood handling 80-95% zero entries.
Tree & Ensemble Methods for Prediction
Rust ML toolkit inspired by scikit-learn with GLMs, clustering (K-Means), PCA, SVM, and regularization (Lasso/Ridge).
Gradient boosting library excelling with categorical features (minimal preprocessing needed). Robust against overfitting.
Fast, distributed gradient boosting (also supports RF). Known for speed, low memory usage, and handling large datasets.
Extends gradient boosting to probabilistic prediction, providing uncertainty estimates alongside point predictions. Built on scikit-learn.
(`RandomForestClassifier`/`Regressor`) Widely-used, versatile implementation of Random Forests. Easy API and parallel processing support.
Rust ML library with regression, classification, clustering, matrix decomposition (SVD, PCA), and model selection tools.
High-performance, optimized gradient boosting library (also supports RF). Known for speed, efficiency, and winning competitions.
GPU-accelerated implementation of Random Forests for significant speedups on large datasets. Scikit-learn compatible API.
Academic APIs
LLM-powered research assistant with superhuman performance on scientific Q&A benchmarks. Agentic RAG with iterative query refinement, automatic metadata fetching, and retraction checking.
Python wrapper for the arXiv API to search and download papers from 1M+ articles in physics, math, CS, and related fields. Download PDFs and source files programmatically.
Low-level Python client for CrossRef API to retrieve DOI metadata, citations, and bibliographic data. Supports polite pool access for faster response times.
R interface to OpenAlex, the free open catalog of 240M+ scholarly works. Query publications, authors, institutions, and citation networks without subscription database access.
Lightweight Python interface to OpenAlex API for querying 240M+ scholarly works, authors, institutions, and topics. Supports pagination and converts inverted abstracts to plaintext.
Time Series Forecasting
Time series forecasting and analysis for Rust with ETS, MSTL decomposition, seasonality detection, outlier detection, and Prophet-style models.
A tidyverse-native forecasting framework providing ETS, ARIMA, and other models for tidy time series (tsibble objects). Enables fitting multiple models across many time series simultaneously with a consistent formula-based interface.
The foundational R package for univariate time series forecasting. Provides methods for exponential smoothing via state space models (ETS), automatic ARIMA modeling with auto.arima(), TBATS for complex seasonality, and comprehensive model evaluation tools.
Automatic forecasting procedure based on an additive decomposable model with non-linear trends, yearly/weekly/daily seasonality, and holiday effects. Robust to missing data, trend shifts, and outliers; designed for business time series with strong seasonal patterns.
Scalable time series forecasting using machine learning models (e.g., LightGBM, XGBoost) as regressors.
Deep learning models (N-BEATS, N-HiTS, Transformers, RNNs) for time series forecasting, built on PyTorch Lightning.
Forecasting procedure for time series with strong seasonality and trend components, developed by Facebook.
Fast, scalable implementations of popular statistical forecasting models (ETS, ARIMA, Theta, etc.) optimized for performance.
ARIMA modeling with automatic parameter selection (auto-ARIMA), similar to R's `forecast::auto.arima`.
Unified framework for various time series tasks, including forecasting with classical, ML, and deep learning models.
Bayesian Causal Inference
Bayesian Structural Time Series providing the foundation for CausalImpact. Supports spike-and-slab variable selection, multiple state components (trend, seasonality, regression), and non-Gaussian outcomes. Developed at Google.
Bayesian Additive Regression Trees for causal inference. Strong ACIC competition performer with sparsity-inducing priors for multilevel/grouped data.
Causal Inference (Bounds)
Implements modern treatment effect bounds beyond basic Manski worst-case scenarios. Provides tighter bounds using monotonicity, mean independence, and other assumptions following Lee and Weidner (2021).
Core Libraries & Linear Models
Simple linear regression for Rust with R-style formula syntax, standard errors, t-stats, and p-values.
Blazingly fast DataFrame library for Rust and Python with SQL-like syntax, lazy evaluation, and excellent time series handling.
Applied Econometrics Library bridging Stata-like syntax with Python. Built on statsmodels with convenient API.
H2O's distributed ML engine on Spark with GLM/GAM that provides p-values, confidence intervals, and Tweedie/Gamma distributions.
Foundational ML library with regression models (incl. regularized), model selection, cross-validation, evaluation metrics.
Comprehensive library for estimating statistical models (OLS, GLM, etc.), conducting tests, and data exploration. Core tool.
Energy Systems Modeling
Uplift Modeling
Uplift modeling for observational (non-RCT) data using inverse probability weighting.
Booking.com's enterprise uplift modeling via PySpark and H2O. Six meta-learners plus Uplift Random Forest with ROI-constrained optimization.
Wayfair's uplift modeling wrapping sklearn for speed with rigorous Qini curve evaluation.
Marginal Effects
Estimated Marginal Means (least-squares means) for factorial designs. Computes adjusted means and contrasts for balanced and unbalanced designs, with support for mixed models and Bayesian models.
Modern standard for interpreting regression results—up to 1000× faster than margins. Computes marginal effects, predictions, contrasts, and slopes for 100+ model classes. Published in JSS 2024.
Matching & Market Design
Python O(n²) implementation of Gale-Shapley algorithm for stable matching with simulation capabilities.
Implements Stable Marriage, Hospital-Resident, Student-Allocation, and Stable Roommates using Gale-Shapley (JOSS paper).
Student-Project Allocation with lecturer preferences. Extends matching to three-sided markets.
Neural network optimal auction design. Implements RegretNet, RochetNet for mechanism design.
Kidney exchange optimization with hierarchical objectives. Production-ready for kidney paired donation.
R/C++ implementation of Gale-Shapley and Irving's algorithms for stable matching. Tested with 30,000+ participants.
Matching with couples using Scarf's algorithm. Essential for NRMP-style medical residency matching.
Visualization
Compose multiple ggplot2 plots into publication-ready multi-panel figures. Uses intuitive operators (+, |, /) for arrangement with automatic alignment and shared legends.
Publication-ready ggplot2 themes and plot arrangement utilities. Provides clean themes, plot annotations, and functions for combining plots with shared axes.
Robust Standard Errors
Provides cluster-robust variance estimators with small-sample corrections, including bias-reduced linearization (BRL/CR2). Includes functions for hypothesis testing with Satterthwaite degrees of freedom and Hotelling's T² approximation—essential when the number of clusters is small.
Collection of tests for diagnostic checking in linear regression models. Provides the essential coeftest() function for testing coefficients with alternative variance-covariance matrices (pairs with sandwich), plus Breusch-Pagan, Durbin-Watson, and RESET tests.
Object-oriented software for model-robust covariance matrix estimators including heteroscedasticity-consistent (HC0-HC5), heteroscedasticity- and autocorrelation-consistent (HAC/Newey-West), clustered, panel, and bootstrap covariances. Works with lm, glm, fixest, survival models, and many others.
Sports Analytics
R package for NHL play-by-play data with built-in expected goals models and player tracking statistics
Python package for accessing nflverse NFL play-by-play data with built-in EPA and win probability models
R package providing the complete Lahman Baseball Database as native R data frames for seamless analysis
R package for accessing NBA Stats API plus ESPN and KenPom data for comprehensive basketball analytics
Python library for football/soccer pitch visualization with support for heat maps, shot maps, pass maps, and event plotting
Full NBA Stats API wrapper with 127+ endpoints for accessing shot charts, player tracking, play-by-play, and historical data
R package for NFL play-by-play data with built-in expected points (EPA) and win probability models from 1999-present
Python library for pulling baseball data from Statcast, FanGraphs, Baseball Reference, and the Lahman database with easy-to-use functions
Official Python API client for StatsBomb open data with 360 freeze-frame support for detailed soccer event analysis
R package for scraping FBref, Transfermarkt, and Understat soccer data including xG, player values, and match statistics
Model Diagnostics
Functions accompanying 'An R Companion to Applied Regression.' Provides advanced regression diagnostics including variance inflation factors (VIF), Type II/III ANOVA, influence measures, linear hypothesis testing, power transformations (Box-Cox), and comprehensive diagnostic plots.
Utilities for computing indices of model quality and goodness of fit, including R², RMSE, ICC, AIC/BIC. Provides functions to check models for overdispersion, zero-inflation, multicollinearity (VIF), convergence, and singularity. Supports mixed effects and Bayesian models.
Visualization toolbox for the easystats ecosystem built on ggplot2. Provides publication-ready plotting methods for model parameters, predictions, and performance diagnostics from all easystats packages via simple plot() calls.
Mixed Effects
Fit generalized linear mixed models with extensions including zero-inflation, hurdle models, heteroscedasticity, and autocorrelation using Template Model Builder (TMB) with automatic differentiation and Laplace approximation.
Fit linear and generalized linear mixed-effects models using S4 classes with Eigen C++ library for efficient computation, supporting arbitrarily nested and crossed random effects structures for hierarchical and longitudinal data.
Provides p-values for lme4 model fits via Satterthwaite's or Kenward-Roger degrees of freedom methods, with Type I/II/III ANOVA tables, model selection tools (step, drop1), and least-squares means calculations.
Fit Gaussian linear and nonlinear mixed-effects models with flexible correlation structures, variance functions for heteroscedasticity, and nested random effects. Ships with base R and offers more variance-covariance flexibility than lme4.
Power Analysis
Comprehensive collection of tools for basic and advanced statistical power analysis including correlation, t-test, ANOVA, regression, mediation analysis, structural equation modeling (SEM), and multilevel models. Features both R package and web interface.
Provides basic power calculations using effect sizes and notation from Cohen (1988). Supports t-tests, chi-squared tests, one-way ANOVA, correlation tests, proportion tests, and general linear models with analytical (closed-form) solutions.
Calculates power for generalized linear mixed models (GLMMs) using Monte Carlo simulation. Designed to work with lme4 models; supports LMMs and GLMMs with crossed random effects, non-normal responses, and complex variance structures where analytical solutions are unavailable.
Spatial Econometrics
The modern standard for spatial vector data in R, implementing Simple Features access (ISO 19125). Represents spatial data as data frames with geometry list-columns, enabling seamless tidyverse integration. Interfaces with GDAL (I/O), GEOS (geometry operations), PROJ (projections), and s2 (spherical geometry).
The foundational R package for spatial weights matrix creation and spatial autocorrelation testing. Provides functions for creating spatial weights from polygon contiguities and point patterns, computing global statistics (Moran's I, Geary's C), local indicators (LISA), and Lagrange multiplier tests.
The broader PySAL ecosystem contains many tools for spatial data handling, weights, visualization, and analysis.
Distributed spatial analytics engine (formerly GeoSpark) with spatial SQL, K-NN joins, and range queries for spatial econometrics.
The spatial regression `spreg` module of PySAL. Implements spatial lag, error, IV models, and diagnostics.
Comprehensive package for spatial regression model estimation, split from spdep in 2019. Provides maximum likelihood, two-stage least squares, and GMM estimation for spatial lag (SAR), spatial error (SEM), and combined (SARAR/SAC) models, plus Spatial Durbin and SLX variants with impact calculations.
Maximum likelihood and GMM estimation for spatial panel data models. Implements fixed and random effects specifications with spatial lag and/or spatial error components, including the Kapoor-Kelejian-Prucha (2007) GM estimator. Provides diagnostic tests for spatial autocorrelation in panel settings.
Synthetic Data Generation
Privacy-preserving synthetic data using Bayesian networks with differential privacy. From University of Washington DataResponsibly project.
Time series synthetic data generation using deep learning. Part of the SDV ecosystem for sequential data.
GAN-based tabular data synthesizer using Variational GMM for mode-specific normalization. Published at NeurIPS 2019. Core component of SDV ecosystem.
Comprehensive fake data generator for 50+ locales including names, addresses, financial data, and more. Most popular Python library for test data generation.
Open-source synthetic data library with DGAN for time series, ACTGAN, and differential privacy support from Gretel.ai.
High-performance fake data generator—faster than Faker. Provides data for multiple domains and 35+ locales.
Comprehensive library for generating synthetic tabular, relational, and time series data using various models.
Port of the R package for generating synthetic populations based on sample survey data.
Statistical Disclosure Control for microdata used by World Bank and census agencies. Comprehensive anonymization toolkit.
Synthetic population simulation for EU-SILC style survey data. Creates realistic household and individual-level synthetic populations.
Survival Analysis
PyTorch-based survival analysis. Implements DeepSurv, DeepHit, Cox-Time, and other neural survival models with partial likelihood and direct prediction approaches.
CTR Prediction
Easy-to-use implementations of deep CTR models including Wide&Deep, DeepFM, DIN, xDeepFM, and multi-task architectures
Causal Inference (Continuous Treatment)
Machine learning-based generalized propensity score estimation for continuous treatments. Uses SuperLearner ensemble methods for flexible estimation of dose-response curves.
Causal Inference (Interference)
Computes inverse probability weighted (IPW) causal effects under partial interference following Tchetgen Tchetgen and VanderWeele (2012). Handles spillover effects within groups while maintaining independence across groups.
Handles both noncompliance AND network interference of unknown form following Hoshino and Yanagi (2023 JASA). Provides valid inference when treatment effects spill over through network connections.
Causal Inference (Event Study)
Implements event study best practices from Freyaldenhoven et al. (2021) including sup-t confidence bands for uniform inference and formal pre-trend testing. Provides robust methods for dynamic treatment effect estimation.
Streamlined event study workflows with simple run_es() and plot_es() functions built on fixest. New 2025 package providing convenient wrappers for common event study specifications.
Causal Inference (Dynamic Treatment)
Dynamic treatment regime estimation via G-estimation for sequential treatment decisions. Implements methods for finding optimal treatment rules that adapt over time based on patient characteristics.
Comprehensive package for dynamic treatment regimes implementing Q-learning, value search, and outcome-weighted learning methods. Accompanies the textbook 'Dynamic Treatment Regimes' (Tsiatis et al., 2020).
Generalized Additive Models
Distributional regression where all parameters of a response distribution (location, scale, shape) can be modeled as functions of predictors, supporting 100+ distributions including highly skewed and kurtotic continuous and discrete distributions.
The definitive GAM implementation providing generalized additive (mixed) models with automatic smoothness estimation via REML/GCV/ML, supporting thin plate splines, tensor products, multiple distributions, and scalable fitting for large datasets.
Game Theory & Mechanism Design
Computation of Nash equilibria for 2-player games. Support enumeration and Lemke-Howson algorithm.
DeepMind's 70+ game environments with multi-agent RL algorithms including Alpha-Rank, Neural Fictitious Self-Play, and CFR variants.
Fair division algorithms from academic papers. Implements cake-cutting and item allocation procedures.
Course-seat allocation with capacity constraints. Practical fair division for university course assignment.
N-player extensive form games with Alan Turing Institute support. Computes Nash, perfect, and sequential equilibria.
Causal Inference (Mediation)
Unified interface for six causal mediation approaches including traditional regression, inverse odds weighting, and g-formula. Supports multiple sequential mediators and exposure-mediator interactions.
Estimates Average Causal Mediation Effects (ACME) with sensitivity analysis for unmeasured confounding. Implements Tingley et al. (2014 JSS) methods for understanding causal mechanisms.
Datasets
Companion package to 'Applied Econometrics with R' (Kleiber & Zeileis) plus datasets from Stock & Watson. Provides ivreg() for instrumental variables, tobit(), and econometric testing functions.
Unified collection of datasets from three major causal inference textbooks: 'The Effect' (Huntington-Klein), 'Causal Inference: The Mixtape' (Cunningham), and 'Causal Inference: What If?' (Hernán & Robins).
All 115 datasets from Wooldridge's 'Introductory Econometrics: A Modern Approach' (7th edition). Includes wage equations, crime data, housing prices, and classic econometrics teaching examples.
Causal Inference (Principal Stratification)
Principal stratification analysis for noncompliance and truncation-by-death using both Bayesian (Stan) and frequentist estimation. Implements Liu and Li (2023) methods for causal inference with post-treatment complications.
Conjoint Analysis
Estimates Average Marginal Component Effects (AMCEs) for conjoint experiments following Hainmueller, Hopkins & Yamamoto (2014). Handles multi-dimensional preferences with clustered standard errors.
Tidy interface for conjoint analysis with visualization. Provides functions for calculating and plotting marginal means and AMCEs with ggplot2-based output for publication-ready figures.
Data Workflow
High-performance data transformation package designed by an economist. Provides fast grouped operations, time series functions, and panel data tools with 10-100× speedups over dplyr on large data.
Extension of data.frame providing fast aggregation of large data (100GB+), ordered joins, and memory-efficient operations. Uses reference semantics for in-place modification with concise syntax [:=, .SD, by=].
Import and export Stata, SPSS, and SAS data files preserving variable labels and value labels. Handles .dta, .sav, .sas7bdat, and .xpt formats with labelled vectors for metadata.
Meta-package installing core tidyverse packages: ggplot2 (visualization), dplyr (manipulation), tidyr (tidying), readr (import), purrr (functional programming), tibble (data frames), stringr (strings), and forcats (factors).
Dimensionality Reduction
Specialized library for Exploratory (EFA) and Confirmatory (CFA) Factor Analysis with rotation options for interpretability.
Optimized, parallel implementation of t-distributed Stochastic Neighbor Embedding (t-SNE) for large datasets.
Fast and scalable implementation of Uniform Manifold Approximation and Projection (UMAP) for non-linear reduction.
Cybersecurity
Python wrapper for the NIST National Vulnerability Database (NVD) API for automated vulnerability intelligence
Marketing Mix Models (MMM) & Business Analytics
Google's Zero-Inflated Lognormal loss for heavily-tailed LTV distributions. Outputs both predicted LTV and churn probability.
Analyze customer lifetime value (CLV) using probabilistic models (BG/NBD, Pareto/NBD) to predict purchases.
Lightweight Python library focused specifically on Marketing Mix Modeling implementation.
Collection of Bayesian marketing models built with PyMC, including MMM, CLV, and attribution.
Defense Research
Python package for conflict event data visualization and geospatial analysis
R package for generating dyad-year and state-year datasets with conflict, democracy, alliance, and contiguity data
Marketing Analytics
Meta's AI/ML-powered Marketing Mix Modeling package with ridge regression and multi-objective optimization
Bayesian Marketing Mix Modeling and Customer Lifetime Value with PyMC, including GPU acceleration
Data Wrangling
R package for converting between country naming and coding conventions essential for merging defense datasets
Interference & Spillovers
Meta's library for estimating heterogeneous spillover effects in A/B tests. Handles network interference.
Treatment and spillover effect estimation under network interference. Separates direct and indirect effects.
Statistical tests for SUTVA violations and spillover hypotheses. Detects network interference in experiments.
Inference & Reporting Tools
Open-source textbook by Richard Evans on computational methods for researchers using Python.
Publication-ready regression tables for pyfixest, statsmodels, linearmodels. Outputs HTML (great-tables), LaTeX, Word.
Cookiecutter templates for reproducible economics research projects. Standardized project structure.
Curated collection of Python packages for applied researchers organized by functionality.
Wild cluster bootstrap and pairs cluster bootstrap implementations for clustered standard errors.
Text Analysis
Structural Topic Models incorporating document-level metadata as covariates affecting topic prevalence and content. Enables studying how topics vary across groups or time with uncertainty quantification.
Comprehensive framework for quantitative text analysis. Provides fast text preprocessing, document-feature matrices, dictionary analysis, and integration with topic models. Standard for political science text analysis.
Efficient text vectorization with word embeddings (GloVe), topic models (LDA), and document similarity. Memory-efficient streaming API for large corpora with C++ backend.
Tidy data principles for text mining. Converts text to tidy format (one-token-per-row), enabling analysis with dplyr, ggplot2, and other tidyverse tools. Accompanies the book 'Text Mining with R'.
Standard Errors, Bootstrapping & Reporting
Curated list of quantitative finance libraries and resources (many statistical/TS tools overlap with econometrics).
Teaches software design principles for ML—modularity, abstraction, and reproducibility—going beyond ad hoc Jupyter workflows. Focus on maintainable, production-quality ML code.
Modern introduction to causal inference methods (DiD, IV, RDD, Synth, ML-based) with Python code examples.
Practical guide by A. Turrell on using Python for modern econometric research, data analysis, and workflows.
Intermediate 5-course series by Andrew Ng covering deep neural networks, CNNs, RNNs, transformers, and real-world DL applications using TensorFlow.
Beginner-friendly 3-course series by Andrew Ng covering core ML methods (regression, classification, clustering, trees, NN) with hands-on projects.
Comprehensive intro notes by Kevin Sheppard covering Python basics, core libraries, and econometrics applications.
High-quality lecture series on quantitative economic modeling, computational tools, and economics using Python/Julia.
(`scipy.stats.bootstrap`) Computes bootstrap confidence intervals for various statistics using percentile, BCa methods.
Python port of R's stargazer for creating publication-quality regression tables (HTML, LaTeX) from `statsmodels` & `linearmodels` results.
Teaches essential developer tools often skipped in formal education—command line, Git, Vim, scripting, debugging, etc.
Fast implementation of various wild cluster bootstrap algorithms (WCR, WCU) for robust inference, especially with few clusters.
Structural Equation Modeling
Extended SEM software with programmatic model specification via paths (RAM) or matrix algebra, supporting mixture distributions, item factor analysis, state space models, and behavior genetics twin studies.
Bayesian latent variable analysis extending lavaan with MCMC estimation via Stan or JAGS, supporting Bayesian CFA, SEM, growth models, and model comparison with WAIC, LOO, and Bayes factors.
Free, open-source latent variable analysis providing commercial-quality functionality for path analysis, confirmatory factor analysis, structural equation modeling, and growth curve models with intuitive model syntax.
Space & Orbital Analysis
Elegant astronomy library for computing satellite and celestial positions using JPL ephemeris data
Implementation of the SGP4/SDP4 satellite propagation algorithms for processing TLE orbital data
Python client for the Space-Track.org API to access satellite catalog and TLE data
Quantile Regression & Distributional Methods
Fast quantile regression solver using interior point methods, supporting robust and clustered standard errors.
Recentered Influence‑Function (RIF) regression for unconditional quantile & distributional effects (Firpo et al., 2008).
Scikit-learn compatible implementation of Quantile Regression Forests for non-parametric estimation.
Network Analysis
Grammar of graphics for network data built on ggplot2. Provides layouts, geometries, and faceting specifically designed for network visualization with publication-quality output.
Comprehensive network analysis library with efficient algorithms for network creation, manipulation, and analysis. Provides centrality measures, community detection, graph visualization, and network statistics.
Community detection in large networks using the Louvain algorithm, applicable to defense network analysis
Social network analysis tools including network visualization, centrality measures, and statistical models for network data. Part of the statnet suite for network regression and exponential random graph models.
Tidy data interface for network/graph data. Extends dplyr verbs to work with nodes and edges, enabling pipe-friendly network manipulation that integrates seamlessly with ggraph for visualization.
Research Tools
Visual tool for exploring academic paper relationships. Creates visual graphs showing prior and derivative works.
AI-powered academic search engine providing evidence-based answers from peer-reviewed literature with economics specialty.
AI research assistant that automates literature review across 126M+ academic papers. 99%+ accuracy in data extraction from research papers.
Free tool for discovering academic papers through network visualization of paper connections and co-authorships.
AI-powered research tool with 200M+ papers indexed. Free API access for academic paper search and citation analysis.
Regression Output
Converts messy output from 100+ statistical model types into consistent tidy tibbles using three verbs: tidy() for coefficient-level statistics, glance() for model-level summaries (R², AIC), and augment() for fitted values and residuals.
Build display tables from tabular data using a cohesive grammar of table parts (header, stub, body, footer). Enables progressive construction of publication-quality tables with extensive formatting, footnotes, and cell styling. Outputs to HTML, LaTeX, and RTF.
Creates publication-ready analytical and summary tables (Table 1 demographics, regression results, survival analyses) with one line of code. Auto-detects variable types, calculates appropriate statistics, and formats regression models with reference rows and appropriate headers.
Creates publication-quality tables summarizing multiple statistical models side-by-side, plus coefficient plots, data summaries, and correlation matrices. Supports 100+ model types via broom/parameters with output to HTML, LaTeX, Word, PDF, PNG, and Excel.
Produces well-formatted LaTeX, HTML/CSS, and ASCII regression tables with multiple models side-by-side, plus summary statistics tables. Widely used in economics with journal-specific formatting styles (AER, QJE, ASR).
Converts coefficients, standard errors, significance stars, and fit statistics from statistical models into LaTeX, HTML, Word, or console output. Highly extensible with support for custom model types and confidence intervals.
Reproducibility
Simple path construction from project root. Uses heuristics to find project root (RStudio, .git, .here) enabling portable paths that work across different machines and working directories.
Project-local R dependency management. Creates reproducible environments by recording package versions in a lockfile, isolating project libraries, and enabling version restore.
Dynamic documents combining R code with Markdown text. Generates reproducible reports in HTML, PDF, Word, and slides. Foundation for literate programming and reproducible research in R.
Make-like pipeline toolkit for R. Declares dependencies between pipeline steps, skips up-to-date targets, and supports parallel execution. Standard for reproducible research workflows.
Geospatial
Recommender Systems
A Python scikit for building and analyzing recommender systems with explicit ratings. Implements SVD, SVD++, NMF, k-NN, and other classic collaborative filtering algorithms. The go-to library for Netflix Prize-style recommendations.
PDF & Document Processing
Machine learning library for extracting structured data from scholarly PDFs. Parses headers, references, authors, affiliations with ~0.87-0.90 F1-score. Used by ResearchGate, Semantic Scholar, Internet Archive.
Convert PDF, DOCX, PPTX to markdown with high accuracy. Handles tables, equations (LaTeX), code blocks, and references. 10x faster than Nougat with optional LLM enhancement for complex layouts.
High-performance Python library for PDF data extraction, analysis, and manipulation. Processes 36 documents/second for headers with excellent text fidelity. Wraps the MuPDF library.
Plumb PDFs for detailed information about characters, rectangles, lines, and tables. Excels at table extraction with visual debugging tools. Built on pdfminer.six.
Extract text, metadata, and positional data from PDF documents in R. Part of rOpenSci, processes PDFs at ~0.1 seconds per document with excellent text fidelity.
Bibliography Management
Parse and write BibTeX files in Python with middleware transformations. Trusted by 1,600+ repositories. Version 2.0 beta offers 10x faster performance.
Powerful CLI document and bibliography manager with git-like interface. Auto-fetches metadata from DOI/arXiv/ISBN, exports to BibTeX, integrates with Vim/Emacs. Syncs via git/Dropbox.
Comprehensive bibliography management in R with BibTeX/BibLaTeX read/write, CrossRef and Zotero API integration, UTF-8 support, and RMarkdown citation generation.
Comprehensive science mapping and bibliometric analysis. Imports from Web of Science, Scopus, PubMed, OpenAlex. Provides co-citation networks, thematic evolution, and the Biblioshiny GUI. Over 5,200 academic citations.
Convert between 15+ bibliography formats including BibTeX, EndNote, RIS, MODS XML, and more. Low-level format conversion for academic reference management.
Geospatial & Spatial Economics
Python package integrating AI with geospatial data analysis and visualization. 70+ end-to-end Jupyter notebook examples and a QGIS plugin. Published in JOSS.











































































































