picasso is a model that allows making predictions for the thermodynamic properties of the gas in massive dark matter halos from gravity-only cosmological simulations. It combines an analytical model of gas properties as a function of gravitational potential with a neural network predicting the parameters of said model. It is released here as a Python package, combining an implementation of the gas model based on JAX and Flax, and models that have been pre-trained to reproduce gas properties from hydrodynamic simulations.
10.33232/001c.127486