I am currently writing an model from torch -> jax/flax, chatgpt is pretty bad with it because halucinates whole functions/modules that are non existing. Gemini is better because google uses maybe more internal code to train on ?