Featured work

1. VachanVY/NeuroForge

   Unveiling the Layers: Neural Networks from first principles

   Jupyter Notebook 1
2. VachanVY/gpt.jax

   Generative Pretrained Model (GPT) in JAX. A step by step guide to train LLMs on large datasets from scratch

   Python 2
3. VachanVY/Transfusion.torch

   PyTorch Implementation of Transfusion: Predict the Next Token and Diffuse Images with One Multi-Modal Model

   Python 15
4. VachanVY/diffusion-transformer

   Pytorch and JAX Implementation of Scalable Diffusion Models with Transformers | Diffusion Transformers in Pytorch and JAX

   Python 3
5. VachanVY/Reinforcement-Learning

   || Studying RL the hard way || Implementation of Important Algorithms in PyTorch from "Reinforcement Learning an Introduction" by Sutton and Barto along with RL papers

   Python 1