The GitHub repository for the paper "JAT" accepted by NeurIPS 2022. - View it on GitHub