Good to hear it’s working!
It was a good idea to compare both models with a constant initialization scheme, as both frameworks will most likely return different “random” values even for the same seed.