Any publicly available document?

[Lekja00160612]

Hi Praxis team, I have been using Jax and Flax for quite sometime before find out Praxis. Flax was great, however, not well suited for scaling. I also checked T5X and Flaxformer but it seems like they are not very developer friendly as the main functionality is defining transformer layers. Me and my collegues would love to move to Praxis in our future work. I wonder if there are any available document that we can use?

[zhangqiaorjc]

Sorry for the late reply. The README in

https://github.com/google/paxml/tree/main
https://github.com/google/paxml/tree/main/paxml/contrib/gpu

are probably the best public document now.