Contribute to Implement open LLM models with JAX and Flax

[YuvrajSingh-mist]

Hi @SamanehSaadat @divyashreepathihalli

Excited to work on one of the ideas called Implement open LLM models with JAX and Flax
.

I have relevant experience in implementing LLM models from scratch and training them (pre + pos).

Do let me know if I can get started right away!

Links to my projects -

https://github.com/YuvrajSingh-mist/Paper-Replications/tree/master

https://github.com/YuvrajSingh-mist/SmolLlama

https://github.com/YuvrajSingh-mist/StoryLlama/tree/master