Rohit Bele · Machine Learning Engineer

Understanding intelligence by building it.

I work on machine learning systems, from implementing neural networks and automatic differentiation to reproducing sequence models and studying how models train, reason, and generalize.

Exploring: Deep Learning · LLMs · Training Dynamics · RLHF

Featured work

3 case studies
CompleteGitHub ↗

Seq2Seq from Scratch

A practical-scale reproduction of the 2014 sequence-to-sequence architecture, with explicit LSTM gates, controlled source-reversal experiments, and deliberately reported failure modes.

Four small-scale CPU runs reported test BLEU of 1.248–2.715. Source reversal did not consistently help.

  • Python
  • NumPy
  • PyTorch
  • LSTM
  • NLP
  • SacreBLEU
CompleteGitHub ↗

Micrograd / Automatic Differentiation

Learning extensions to Andrej Karpathy's micrograd, with input-shape fixes, gradient regression tests, a backpropagation walkthrough, and reproducible command-line training.

Tests cover shared operands and gradient accumulation. The scalar engine has no tensors or GPU acceleration.

  • Python
  • Autograd
  • Backpropagation
  • Pytest
  • Graphviz
  • GitHub Actions
CompleteGitHub ↗

Character-Level RNN

A NumPy-only character language model that implements the recurrent forward pass, BPTT, gradient checking, Adam, checkpointing, and controlled sampling by hand.

The continuation run reported 2.1348 validation loss per character; generated text still lacks consistent coherence.

  • Python
  • NumPy
  • RNN
  • BPTT
  • Adam
  • Pytest
View all projects

Latest note

#019

Why Learning Rates Should Decay During Training

Learning-rate decay reduces the effect of noisy late-stage updates while preserving larger steps early in training.

Machine Learning8 min read

Read note

More notes

All 19 notes