Check our papers Safe control under input saturation with neural control barrier functions. ARC - Actor Residual Critic for Adversarial Imitation Learning.