Featured work

1. VachanVY/NeuroForge

   Unveiling the Layers: Neural Networks from first principles

   Jupyter Notebook 10
2. VachanVY/gpt.jax

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

   Python 4
3. VachanVY/Transfusion.torch

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

   Python 25
4. VachanVY/diffusion-transformer

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

   Python 8
5. VachanVY/Reinforcement-Learning

   PyTorch implementations of algorithms from "Reinforcement Learning: An Introduction by Sutton and Barto", along with various RL research papers.

   Python 198