Machine Learning Engineer

Hi, I'm Afshin Shahrestani.

Machine Learning Engineer specializing in computer vision and large-scale data pipelines. Built zero-to-one CV/ML systems across GCP/AWS, including transformer-based grounding, detection, and segmentation workflows. Strong in Python, PyTorch, OpenCV, and high-throughput video/data pipelines.

Summary

From Research To Production Systems.

I own the full ML lifecycle—from data ingestion to models to deployed services—implementing monitoring and reliability practices while collaborating tightly with backend/cloud teams through CI/CD, code reviews, and containerized deployments.

Recent work includes transformer-based grounding, detection, and segmentation workflows using HuggingFace, SAM-style models, and custom pre/post-processing. Strong in Python, PyTorch, OpenCV, Hydra config management, and high-throughput video/data pipelines.

Experience

Impact In Production

Key roles and achievements. Expand for details.

Tools and stack

Tools And Stack

ML Frameworks

  • PyTorch, TensorFlow, scikit-learn
  • HuggingFace Transformers
  • LangChain
  • OpenCV

Computer Vision

  • Vision Transformers (ViT, DETR, DINO)
  • CNN Architectures (ResNet, EfficientNet)
  • Grounding Models (GroundingDINO)
  • Detection & Segmentation (SAM, Detectron2)
  • Structure from Motion (SfM, OpenSfM)

LLMs & NLP

  • Large Language Models (GPT, Llama, Phi-3)
  • Prompt Engineering
  • RAG Pipelines & Vector Search
  • LangChain, FAISS
  • Embeddings & Semantic Search

MLOps & Cloud

  • Docker, GitHub Actions CI/CD
  • GCP (Vertex AI, Cloud Run, GCS)
  • AWS (SageMaker, ECS, S3, Lambda)
  • Hydra, Weights & Biases

Data & Databases

  • MongoDB, MySQL, SQL Server
  • Data Versioning (DVC)
  • ETL & Data Pipelines

Projects

Selected Works

Highlights from recent projects. View all projects →

Better Exam - Accessible Exam Platform

An exam hosting service for the visually impaired using Azure Cognitive Speech Services. Text-to-Speech reads questions; Speech-to-Text captures answers. Ranked top 50 in Azure AI Hackathon 2021.

RAG Document Retrieval System

A RAG pipeline for querying medical patient records using FAISS vector search and HuggingFace LLMs. Features vector-based retrieval, Phi-3-mini question answering, and 11 demo scenarios.

Beyond work

Gaming, Miniatures, And Gymming.

When I'm not coding, you'll find me gaming, building Warhammer and Gunpla miniatures, or gymming. The hobbies page has more details on what keeps me creative and balanced.

Explore hobbies

Contact

Let's Build Something Measurable.

Open to ML engineering roles, technical leadership opportunities, and collaborative projects in AI and MLOps.