FLAX/JAX Learning Resources
Hello all,I have tried my best to organize FLAX/JAX learning resources. I hope it will be useful for all of us.If you have more learning resources, drop a comment & I will add them here…JAXis a numerical computation library that exposes a NumPy-like API with tracing capabilities.Flax is the most widely used JAX library, with129 dependent projectsas of May 2021.Flaxbuilds on top of JAX with an ergonomic module abstraction using Python dataclasses that leads to concise and explicit code.Flax’s “lifted” JAX transformations (e.g.vmap,remat) allow you to nest JAX transformation and modules in any way you wish.  It is also the library underlying all of the official Cloud TPU JAX examples.Github links :JAX ProjectsFLAX examplesColab links:Causal Language Modelling on OSCARMasked Language Modelling on OSCARText Classification on GLUEDocs:JAX 101 TutorialFlax Basics ColabFlax examplesMore useful links:Guide to run JAX on Google Cloud TPU.Load Dataset in streaming modeTALKS Date & Time:30.6.2021  :  9am - 11am PSTZoom link1.7.2021    :  8:30am - 10am PSTZoom link2.7.2021    :  8am - 10am PSTZoom link
2021-06-30T12:15:56Z
Great summary - thanks a lot@srisweet!
2021-06-30T12:23:12Z
My pleasure !@patrickvonplatenCan you please share the links to join the talks ?
2021-06-30T12:28:46Z
There are online and accesable here:transformers/examples/research_projects/jax-projects at master · huggingface/transformers · GitHub
2021-06-30T12:29:54Z
Thank you very much !@patrickvonplaten
2021-06-30T12:52:35Z
My pleasure@alexlau.Patrick will help you with this question.@patrickvonplatenCan you tell us if the zoom talks will be recorded ?
2021-07-01T01:52:19Z
Hi@alexlau,Just found out that the zoom talks have been recorded…I will post the links once I get them…