Jaskaran Virdi Senior ML Engineer
Taking contract and full-time work now

I take

Senior ML engineer, eight years at Twitter, Rubrik and Imprivata. I get AI prototypes deployed, measured, monitored and cheap to run.

Jaskaran Virdi
Previously at
TwitterTrust and Safety
Ransomware detection
Healthcare and identity
AppleTime series forecasting

What I can build for you

Pick the problem. I will take it to production.

I can build the model, the platform under it, or both, on my own or inside your team. Each card shows where I have done the work before.

  1. PrototypeYour notebook, demo or agent
  2. EvaluateEval sets and a quality bar
  3. ShipAPIs, workflows and deployment
  4. MonitorTracing, alerts and drift checks
  5. OptimizeInference cost and speed

Evals and monitoring

Eval sets and regression checks for ML and LLM systems, so you know whether a change made things better or worse. Monitoring by segment, tracing and dashboards once it is live.

Built atTwitter, Imprivata

Deployment and inference cost

Getting models out of notebooks and onto Kubernetes with Argo Workflows or Airflow. Job APIs, async serving, caching, memory tuning and smaller serving fleets, so the cloud bill goes down.

Built atImprivata, Rubrik, Twitter

Fraud, abuse and risk models

Classification and anomaly detection from problem framing to launch. Features, labels, evaluation, bias fixes and alerts that explain which behavior fired them. Including regulated settings like healthcare and identity.

Built atTwitter, Rubrik, Imprivata

Voice AI apps and agents

Real-time voice with ElevenLabs text to speech and Deepgram speech to text, streamed so the reply starts speaking before the model finishes writing.

Built inOwn products

Systems I've built

Where I've done this before

Eight years at Twitter, Rubrik and Imprivata, plus an internship at Apple. Each in a domain where a wrong answer costs money or trust.

Imprivata
Senior ML Engineer. Insider threat and drug-diversion detection for hospitals.
2023 to 2026
  • Risk scoring platform

    Built the job API and the orchestration workflows behind risk scoring, so other teams could launch scoring for each hospital customer, track it and collect results. Then ran the platform in production at scale and fixed the jobs that kept running out of memory.

  • AI assistant retrieval

    A customer-facing assistant answers from company documents, with no personal data in the index.

Rubrik
Senior Software Engineer, ML. Ransomware detection.
2021 to 2022
  • Ransomware detection

    Rebuilt from scratch. Customers got far fewer false alerts, and each alert shows why it fired.

  • Faster predictions

    Same detection, faster responses, and less load on storage.

Twitter
Software Engineer, ML. Trust and Safety, abuse detection.
2018 to 2021
  • Toxicity detection

    Built and shipped models that detect toxic tweets across many languages and countries.

  • Model quality and cost

    Fixed a training loop so the models learned from real reviews instead of their own output. Corrected a bias in the training data and added evaluation by user segment. Optimized a model to use a much smaller serving fleet.

Apple
Machine Learning Intern. Internet Services.
2017
  • Forecasting

    Forecasting for iCloud and related web services.

Earlier: ML researcher at the Contextual Robotics Lab at UC San Diego, software engineer at Zomato.

Ways to work together

Start small or go all in

Remote, or on-site in San Francisco.

1 to 2 weeks

Production readiness review

I go through your prototype, data, evals and infrastructure, then hand you a written plan to ship it.

4 to 10 weeks

Build sprint

I build and ship one system end to end. For instance an AI agent, a RAG feature or a model platform.

Monthly

Ongoing contract

I join your team part-time or full-time and own a piece of the ML or AI roadmap.

Permanent

Full-time

Senior ML and AI engineering roles across LLM apps, agents, modeling and ML infrastructure.

  1. Email me a few lines on what you need built.
  2. We get on a call and go through the problem.
  3. I tell you how I'd build it and what it takes.

Tools I use

Stack

The tools I build with, grouped by the kind of work.

ML and modeling

PyTorchscikit-learnXGBoostAnomaly detectionFraud and risk modelsModel evaluation

LLM and voice AI

AI agentsTool callingLLM evalsOpenAIAnthropic ClaudeGeminiRAGAWS BedrockElevenLabsDeepgramText to speechSpeech to textVercel AI SDK

ML platform and cloud

KubernetesArgo WorkflowsAirflowMetaflowDockerTerraformAWSGCP

Data, serving and monitoring

KafkaClickHouseFastAPIgRPCOpenTelemetryGrafana

Languages

PythonTypeScriptGoC++ScalaSQLNext.js

Education

M.S., Computer Vision and Machine Learning
UC San Diego · Summa cum laude
B.S., Computer Science
IIT Ropar
Patent · Co-inventor Methods and systems for medical device alarm management Imprivata · Published May 2026 View on Google Patents ↗

Have an AI prototype that needs to ship?

Send me a few lines about what you have built so far and where it is stuck. Email works best. LinkedIn works too.

Email me LinkedIn