Graduating April 2026 • Open to Full-Time ML/Software Engineering Roles Let's Connect →
Applied AI & ML Engineering

Anurag Jain — Building ML Systems That Ship

  • Production ML Pipelines Evaluation‑first workflows, observability, and guardrails.
  • Generative & Multimodal AI Diffusion, LLM alignment, and multimodal interfaces.

About Me

Portrait of Anurag

I'm a Computing Science student at University of Alberta (graduating April 2026) who builds production ML systems and ships real products.

Recently: Built CodeNeuron (40+ active users), achieved 26x accuracy improvement in transformer-based RL research, and deployed full-stack platforms with Docker/GCP. I take ML from research to production.

Open to full-time ML Engineer or Software Engineer roles starting May 2026.

View Resume

Education

2022–2026
BSc Computing Science @ University of Alberta
• Specialization: Artificial Intelligence & Machine Learning
• Key Courses: Deep Learning, Computer Vision, Algorithms, Data Science
• Awards: Dean's List (2024, 2025)

Experience

Undergraduate Researcher @ University of Alberta
ongoing

Research in AI/ML with focus on deep learning, reinforcement learning, and generative models.

  • Implemented transformer-based RL agents for Skat (imperfect-information card game), achieving 26x improvement in move prediction accuracy (3% → 80%) through behavioral cloning
  • Engineered custom MLP policy networks with logit-level action masking, reducing training loss by 58%
  • Built hybrid Java-Python architecture integrating game engine with ML models for real-time decision-making
Deep Learning Reinforcement Learning Generative Models
Creator & Developer @ CodeNeuron
ongoing

Built full-stack ML learning platform serving 40+ active users with automated code evaluation and real-time feedback.

  • Developed browser-based IDE with secure Dockerized sandbox for Python execution, implementing isolated environments for safe user-submitted code
  • Deployed on GCP with CI/CD pipelines (GitHub Actions), automating container orchestration for 30+ ML problems across 9 topics
  • Built automated grading system evaluating ML implementations with instant visual feedback (metrics, plots)
Product React FastAPI Docker
Machine Learning Intern @ PCL Construction
4 mos

Built ML models for analyzing complex BIM data to automate parts of construction planning.

  • Structured high‑dimensional building information (BIM) for downstream ML tasks.
  • Achieved 98.21% training accuracy and 95.17% testing accuracy, with robust F1-scores (Train F1: 0.9239, Test F1: 0.8113).
Machine Learning BIM Python
Software Developer Intern @ PaySpaze
4 mos

Built a cross‑platform app in Flutter/Dart and implemented digital authentication flows.

  • Shipped UI features and improved reliability of auth and session handling.
  • Designed user-centric interfaces in Figma, improving UX for cross-platform mobile applications.
Flutter Dart Authentication

Projects

Vision Transformer & Image Captioning

Implemented a ViT for CIFAR‑10 image classification and built an image captioning pipeline by pairing a pre‑trained ViT with GPT‑2. Achieved over 80 % test accuracy and bridged vision and language with a BLEU score of 0.06 on Flickr8k.

Computer Vision Transformer ViT Image Captioning
GitHub Repository

Object Detection & Segmentation

Trained a U‑Net based model for object detection and semantic segmentation on noisy MNIST digits. Tuned hyper‑parameters on Google Colab, reaching 94 % accuracy, 82 % IoU, and 75 % pixel‑wise precision.

Computer Vision Segmentation U‑Net IoU
GitHub Repository

Generative Models

Designed VAE, DDPM and DDIM models from scratch to generate class‑conditioned FashionMNIST images. Implemented robust training pipelines with model checkpointing and gradient clipping, achieving 86 % classification accuracy on generated samples.

Generative Models VAE DDPM DDIM FashionMNIST
GitHub Repository

Skills

Languages
Python Java C++ Assembly Dart Flutter React
Frameworks
PyTorch TensorFlow scikit‑learn OpenCV
ML Focus
Reinforcement Learning Deep Learning Computer Vision Generative Models Transformers
Tools
Git Docker Linux Figma Colab Jupyter

Contact

If you’d like to collaborate or have questions about my work, feel free to reach out via email or connect on LinkedIn.