Skip to content
ASH-OS 1.0 LTS

Amazon ML Summer School '26 · open to ML roles

Ashutosh Ranjan

I build machine learning systems from the primitives up — a 124M-parameter LLM pretrained from scratch, a Kubernetes control plane that schedules its own training jobs, a coding agent shipped to PyPI.

124M
params pretrained
2.46B
tokens trained on
441
tests in CI
1900
LeetCode rating

01 / Selected work

  1. 01

    AshuGPT

    2026

    Decoder-only LLM, pretrained from scratch

    A 124M-parameter transformer trained end to end in raw PyTorch — RoPE, RMSNorm, SwiGLU, KV-cache attention and a byte-level BPE tokenizer, all hand-written across 6,000 lines. No transformers.AutoModel: every layer implemented and measured.

    • 23.53 validation perplexity on 2.46B tokens of FineWeb-Edu — 20,000 steps, 27 h on a single RTX 2080 Ti at 25K tok/s.
    • Cut peak memory 50.9% with gradient checkpointing, accumulation and fused attention, all gradient-identical; FSDP with CPU offload shards a 1.23B config to 2.04 GB/GPU.
    • Sequence packing reclaimed the 89% of each fine-tuning step lost to padding, for 4.40x supervised throughput; DPO on HH-RLHF moved the win rate from 50% to 57.1%.
    • Served from Hugging Face Spaces behind a streaming, KV-cached FastAPI endpoint at 80 tok/s, under 441 tests on GitHub Actions.
    • PyTorch
    • CUDA
    • DDP / FSDP
    • FastAPI
    • HF Spaces
  2. 02

    AshML

    2026

    Kubernetes-native ML training and serving platform

    A Node.js control plane that schedules training onto real Kubernetes Pods: its own scheduler enforcing quota and placement (and printing every node it rejected), an executor driving Jobs from observed Pod status, and PostgreSQL state behind an append-only event log.

    • Runs are dequeued with SKIP LOCKED, so concurrent workers never collide on the same job.
    • Checkpoints upload to MinIO/S3 presigned and are verified against the bucket before any model version may use them.
    • One PRODUCTION version per model, promoted transactionally, then served behind a weighted canary router.
    • ResNet-18/CIFAR-10 proved the whole path: one epoch in 691 s, 65.59% top-1 re-evaluated from object store — SIGKILL the pod and it resumes weights, optimizer, LR schedule and data order.
    • Node.js
    • Kubernetes
    • PostgreSQL
    • MinIO / S3
    • Docker
    • Prometheus
  3. 03

    AshCode

    2026

    Terminal coding agent, published on PyPI

    A streaming agentic loop exposing 12 tools — file edit/patch with diffs, shell with timeout and scrubbed env, grep/glob, web search — plus MCP servers over stdio and HTTP/SSE, driven by any OpenAI-compatible endpoint.

    • Survives long sessions in a finite context window: history compacts at 80% capacity and stale tool output is pruned.
    • A signature-based detector breaks loops of repeating failed tool calls instead of burning the budget.
    • Six approval policies gate destructive shell and filesystem operations.
    • Shipped to PyPI under 36 tests on GitHub Actions across Python 3.11 to 3.13.
    • Python
    • MCP
    • asyncio
    • GitHub Actions

Also built

  • Builds hierarchical mind maps from raw text with keyword extraction, semantic similarity, SVO relation mining, graph centrality and local-LLM refinement.

    Python · NLP · Graphs
  • Multi-model ensemble forecaster built for the Metaculus bot tournaments.

    Python · LLM ensemble
  • Stock price predictor comparing Transformer, LSTM and GRU heads behind a Streamlit frontend.

    PyTorch · StreamlitLive
  • Local-first job application tracker — filterable table, kanban board and funnel stats. No backend, no accounts, no network calls; it all stays in the browser, under 29 test files.

    React 19 · TypeScript · Vitest
  • AI waste-sorting image classifier — 6th place in the regional Smart India Hackathon.

    TensorFlow · Streamlit
  • Finds legally available full movies on YouTube from trusted channels, filterable by language and genre.

    JavaScript · APIs

02 / Experience

  1. Jun 2026 — Present

    Student

    Amazon ML Summer School · Remote

    • Selected in the top 3,000 of 150,000+ applicants, roughly 2%.
    • Advanced machine learning, deep learning, transformers, LLMs, generative AI, computer vision and NLP, taught directly by Amazon Applied Scientists.
  2. Jun 2026 — Jul 2026

    AI/ML Engineer Intern

    Navriti Technologies · Bengaluru · On-site

    • Built an end-to-end text-to-video pipeline turning a topic or a 2,000-word article into a narrated explainer video: LangGraph orchestration, LangChain and self-hosted vLLM scene scripting, a 22B LTX-2.3 diffusion model, Kokoro TTS, Whisper and FFmpeg.
    • Ran that 22B video model plus a 12B Gemma-3 encoder (~66 GB in bf16) on a single 16 GB T4 via 4-bit NF4 quantization, CPU offload, VAE tiling and staged loading — roughly a 5x cut in peak VRAM.
    • Traced fp16 activation overflows producing NaNs and reworked attention memory use, turning an unstable pipeline into reliable 30-second generation.
    • Engineered the synchronised render — Whisper timestamp alignment, karaoke-style ASS subtitles, scene-wise TTS, FFmpeg composition and 2x upscaling — behind a two-stage LLM workflow emitting Pydantic-validated What/Why/How/Where scripts.
    • Prototyped a text-to-mind-map generator using Open Information Extraction, LLM-based relation scoring and knowledge-graph visualisation.
  3. Dec 2025 — Feb 2026

    AI Engineer Intern

    Yantrapragya Developers · Noida · Remote

    • Trained NLP sentiment models for production feedback, improving F1 by 8% with custom preprocessing.
  4. Feb 2025 — Mar 2025

    Software Engineer Intern

    Edunet Foundation · Remote

    • Architected a full-stack application in a monorepo — React with ShadCN UI on the front end, Node and Express behind it.
    • Hardened the backend with strict Zod schema validation and Prisma ORM for database access.
    • Containerised the app with Docker and deployed the API to AWS EC2.
    • Served static assets from S3 through CloudFront, cutting load time and latency.

03 / About

Computer science undergrad at NIT Sikkim. I like the layer most people import: attention kernels, memory budgets, schedulers, checkpoint integrity. Every number on this page came off a run I can reproduce.

Languages

  • C
  • C++
  • Python
  • JavaScript
  • TypeScript
  • SQL

ML & GenAI

  • PyTorch
  • TensorFlow
  • Transformers
  • Diffusers
  • LangChain
  • LangGraph
  • vLLM
  • RLHF / DPO
  • Quantization
  • DDP / FSDP
  • Whisper

Infrastructure

  • Kubernetes
  • Docker
  • PostgreSQL
  • AWS (EC2, S3, CloudFront)
  • MinIO / S3
  • Prometheus
  • Grafana
  • CI/CD
  • Linux
  • FFmpeg

Web

  • React
  • Next.js
  • Node.js
  • Express
  • Prisma
  • Zod
  • FastAPI
  • Flask
  • Tailwind CSS
  • MongoDB

04 / Contact

Open to ML engineering and research roles, and always happy to talk about training runs, inference budgets or anything built from scratch.

authran.off@gmail.com