Implementation of Kalmanformer, modeling the Kalman gain with a transformer - View it on GitHub