ATM Astronomical Tone Mapper using NN SimCLR Adapted from https://github.com/sthalles/SimCLR (to be further modified)