hyw

> Yes this is a major component of my planned work over the next year, but I expect it will take a while before a prototype is available. Thanks for...

Thank you for your patient reply! I really appreciate your work. I have an additional question: Can XLB run on the latext JAX 4.28? I tried running examples on JAX...

> You can find pseudo-code in https://github.com/google-research/neuralgcm/tree/8cffd3f1da4191c4507e97a1510fe468b7d97ada/neuralgcm/reference_code. We haven't released the full code mostly because the implementation is quite messy and entangled with internal Google tools, but we do plan/hope...