Model Weights, Predictions, and Chains: Lens Modeling of STRIDES Strongly Lensed Quasars with NPE

This repository contains model weights, predictions, and chains used for Figures 5 ,6,8, 9, 11, 12, 13, 14 in Erickson et al. '25. 

For more details on how to use these files, please see this jupyter notebook, which uses these files as input, and shows how to re-create the figures mentioned above. 

- hierarchical_inference.tgz contains MCMC chains for all lens mass population models. 

- model_predictions.tgz contains all neural network predictions (i.e. mass model posteriors)

- test_sets.tgz contains input images/data.tfrecords for simulated verification test sets

- trained_models.tgz contains weights of trained neural networks used to generate predictions