33 citations · 48 across the 4 of their papers we have counts for
4 papers
PyTorch FSDP: Experiences on Scaling Fully Sharded Data Parallel
Yanli Zhao, Andrew Gu, Rohan Varma +15
It is widely acknowledged that large models have the potential to deliver superior performance across a broad range of domains. Despite the remarkable progress made in the field of…
GIGA-Lens: Fast Bayesian Inference for Strong Gravitational Lens Modeling
A. Gu, X. Huang, W. Sheu +12
We present GIGA-Lens: a gradient-informed, GPU-accelerated Bayesian framework for modeling strong gravitational lensing systems, implemented in TensorFlow and JAX. The three compon…
Deep Transfer Learning for Infectious Disease Case Detection Using Electronic Medical Records
Ye Ye, Andrew Gu
During an infectious disease pandemic, it is critical to share electronic medical records or models (learned from these records) across regions. Applying one region's data/model to…
Arithmetic of weighted Catalan numbers
Yibo Gao, Andrew Gu
In this paper, we study arithmetic properties of weighted Catalan numbers. Previously, Postnikov and Sagan found conditions under which the -adic valuations of the weighted Cata…