Code for Gas thermODynamics and Matter distribution using jAX (GODMAX). This is also one of the first implementation of halo model in JAX. Ready to run on GPUs and with differentiable likelihood, so can run HMC sampling. Meant for jointly analyzing shear2pt and shear-y correlations.