dsontag

langchain

🦜🔗 Build context-aware reasoning applications

llama

Inference code for LLaMA models

llama_index

LlamaIndex is a data framework for your LLM applications

peft

🤗 PEFT: State-of-the-art Parameter-Efficient Fine-Tuning.

allennlp

An open-source NLP research library, built on PyTorch.

dspy

DSPy: The framework for programming—not prompting—foundation models

NeMo

A scalable generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (Automatic Speech Recognition and Text-to-Speech)

composer

Supercharge Your Model Training

trlx

A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)

alpa

Training and serving large-scale neural networks with auto parallelization.

t5x

RL4LMs

A modular RL library to fine-tune language models to human preferences

NeuroNER

Named-entity recognition using neural networks. Easy-to-use and state-of-the-art results.

smoothquant

[ICML 2023] SmoothQuant: Accurate and Efficient Post-Training Quantization for Large Language Models

torchxrayvision

TorchXRayVision: A library of chest X-ray datasets and models. Classifiers, segmentation, and autoencoders.

truss

The simplest way to serve AI/ML models in production

enformer-pytorch

Implementation of Enformer, Deepmind's attention network for predicting gene expression, in Pytorch

OMOP2OBO

OMOP2OBO: A Python Library for mapping OMOP standardized clinical terminologies to Open Biomedical Ontologies

LiST

Lite Self-Training

primock57

Dataset of 57 mock medical primary care consultations: audio, consultation notes, human utterance-level transcripts.

ETAB

A Benchmark Suite for Visual Representation Learning in Echocardiography

dataset

Multi-LexSum is an abstractive summarization dataset for US Civil Rights Lawsuits

PheValuator

An R package for evaluating phenotype algorithms.

ehr_ml

Code for doing machine learning with various EHRs

Twin_Causal_Nets

Estimating the probabilities of caution via deep monotonic twin networks

cotrain-prompting

Code for co-training large language models (e.g. T0) with smaller ones (e.g. BERT) to boost few-shot performance

weaksup-subset-selection

Subset selection / data pruning for weak supervision

real-time-admissions

Code to accompany paper published in Nature Digital Medicine

parametric-robustness-evaluation

Code for paper "Evaluating Robustness to Dataset Shift via Parametric Robustness Sets"

large-scale-temporal-shift-study

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