NERSC Segmentation integration can be added to this PR if that's the direction we want to go for the demo: https://github.com/als-computing/splash_flows/pull/111