Skip to content
Change the repository type filter

All

    Repositories list

    • Jupyter Notebook
      Creative Commons Zero v1.0 Universal
      21400Updated Oct 14, 2024Oct 14, 2024
    • Python
      0000Updated Sep 14, 2024Sep 14, 2024
    • Code for evaluating single cell foundation models scBERT and scGPT
      Jupyter Notebook
      BSD 3-Clause "New" or "Revised" License
      43020Updated Aug 25, 2024Aug 25, 2024
    • Code for Large-Scale Study of Temporal Shift in Health Insurance Claims. Christina X Ji, Ahmed M Alaa, David Sontag. CHIL, 2023. https://arxiv.org/abs/2305.05087
      Python
      0460Updated Aug 14, 2024Aug 14, 2024
    • Code for Seq-to-Final: A Benchmark for Tuning from Sequential Distributions to a Final Time Point. Christina X Ji, Ahmed M Alaa, David Sontag. https://arxiv.org/abs/2407.09642
      Python
      0000Updated Jul 9, 2024Jul 9, 2024
    • Code for "Finding Regions of Heterogeneity in Decision-Making via Expected Conditional Covariance" at NeurIPS 2021 https://arxiv.org/abs/2110.14508
      Jupyter Notebook
      1300Updated Jun 18, 2024Jun 18, 2024
    • analysis code to reproduce results in NMI submission
      Jupyter Notebook
      BSD 3-Clause "New" or "Revised" License
      1200Updated Jun 1, 2024Jun 1, 2024
    • co-llm

      Public
      Co-LLM: Learning to Decode Collaboratively with Multiple Language Models
      Python
      810200Updated May 7, 2024May 7, 2024
    • Onboarding Humans to work with AI: Algorithms to find regions and describe them in natural language that show how humans should collaborate with AI (NeurIPS23)
      Jupyter Notebook
      31100Updated Mar 15, 2024Mar 15, 2024
    • SCOPE

      Public
      Codebase for SCOPE architecture.
      Python
      2100Updated Feb 7, 2024Feb 7, 2024
    • Human-AI Deferral Evaluation Benchmark (Learning to Defer) AISTATS23
      Python
      11910Updated Jan 8, 2024Jan 8, 2024
    • Code and webpages for our study on teaching humans to defer to an AI
      Jupyter Notebook
      01100Updated Nov 6, 2023Nov 6, 2023
    • TabLLM

      Public
      Python
      MIT License
      43273140Updated Nov 2, 2023Nov 2, 2023
    • Python package for machine learning for healthcare using a OMOP common data model
      Python
      MIT License
      2610210Updated Jun 17, 2023Jun 17, 2023
    • Jupyter Notebook
      MIT License
      0000Updated May 17, 2023May 17, 2023
    • Code for paper, "Falsification of Internal and External Validity in Observational Studies via Conditional Moment Restrictions." This repo is currently under construction. Check back later for end-to-end notebooks recreating the results of our AISTATS 2023 paper.
      Jupyter Notebook
      MIT License
      1200Updated Apr 22, 2023Apr 22, 2023
    • vae_ssl

      Public
      Scalable semi-supervised learning with deep variational autoencoders
      Jupyter Notebook
      MIT License
      2902Updated Feb 10, 2023Feb 10, 2023
    • Code for Sample Efficient Learning of Predictors that Complement Humans (ICML 2022)
      Python
      2500Updated Jan 28, 2023Jan 28, 2023
    • Code for "Consistent Estimators for Learning to Defer to an Expert" (ICML 2020)
      Jupyter Notebook
      MIT License
      611015Updated Jan 28, 2023Jan 28, 2023
    • cfrnet

      Public
      Counterfactual Regression
      Python
      MIT License
      8229430Updated Dec 7, 2022Dec 7, 2022
    • ief

      Public
      Train pharmacodynamic deep generative models to model disease progression
      Jupyter Notebook
      MIT License
      1700Updated Dec 2, 2022Dec 2, 2022
    • ml_mmrf

      Public
      Machine Learning with data from the Multiple Myeloma Research Foundation
      Jupyter Notebook
      MIT License
      0820Updated Dec 2, 2022Dec 2, 2022
    • Code for paper, "Falsification before Extrapolation in Causal Effect Estimation"
      Jupyter Notebook
      MIT License
      2200Updated Sep 27, 2022Sep 27, 2022
    • Code for co-training large language models (e.g. T0) with smaller ones (e.g. BERT) to boost few-shot performance
      Python
      MIT License
      31700Updated Sep 23, 2022Sep 23, 2022
    • Code for paper "Evaluating Robustness to Dataset Shift via Parametric Robustness Sets"
      Python
      MIT License
      4700Updated Jul 6, 2022Jul 6, 2022
    • wilds

      Public
      A machine learning benchmark of in-the-wild distribution shifts, with data loaders, evaluators, and default models.
      Python
      MIT License
      126000Updated Apr 4, 2022Apr 4, 2022
    • prancer

      Public
      Platform enabling Rapid Annotation for Clinical Entity Recognition
      JavaScript
      MIT License
      124918Updated Mar 29, 2022Mar 29, 2022
    • Code for ICML 2021 paper "Regularizing towards Causal Invariance: Linear Models with Proxies" (ICML 2021)
      Jupyter Notebook
      21100Updated Mar 14, 2022Mar 14, 2022
    • Tools to identify and evaluate heterogeneity in decision-making processes.
      Python
      MIT License
      1200Updated Feb 12, 2022Feb 12, 2022
    • Code for Contextual Autocomplete paper published in MLHC2020
      Jupyter Notebook
      21300Updated Aug 7, 2021Aug 7, 2021