Skip to content

Vihitha R

AI/ML Engineer • v**********@gmail.com • +18******876 • drivetube.ai/•••••

Professional Summary

AI/ML Engineer with 4+ years of experience developing machine learning, NLP, and Generative AI applications for consumer technology and enterprise financial environments. Experienced designing recommendation systems, Retrieval-Augmented Generation (RAG) platforms, predictive models and production-grade ML services using PyTorch, Hugging Face, LangChain and cloud MLOps on AWS, Azure and GCP.

Technical Skills

Programming Language: Python,JavaScript,TypeScript,SQL
Backend Technologies: FastAPI
Databases: PostgreSQL,MongoDB,Neo4j
Cloud Platforms: Amazon Web Services,AWS Glue,Kinesis
DevOps & Infrastructure: Docker,Kubernetes
Data Engineering & Processing: Apache Spark,PySpark
Data Analysis & Visualization: Power BI,Tableau
AI/ML Frameworks & Libraries: PyTorch,TensorFlow,scikit-learn,XGBoost,Hugging Face Transformers
Generative AI & LLMs: LangChain,LlamaIndex
Vector Databases & RAG: RAG,Pinecone
MLOps: Amazon SageMaker,MLflow,Feature Engineering
Data Integration & ETL: ETL

Work Experience

Sony, USA
USA
AI/ML Engineer
June 2025 – Present
Worked on AI and personalization for Sony's digital media and streaming products; focused on content discovery, search, and scalable ML services.
Tech Stack: PyTorch, Hugging Face Transformers, LangChain, LlamaIndex, AWS SageMaker, AWS Glue, MLflow, PostgreSQL, Neo4j
  • Engineered personalized recommendation systems using PyTorch and BERT-based content embeddings with Neo4j graph signals and PostgreSQL metadata, increasing content discovery engagement by 18% across streaming services.
  • Fine-tuned transformer models (RoBERTa, LLaMA) with Hugging Face Transformers and PyTorch for multilingual sentiment and intent classification, enabling language-aware personalization across global interaction datasets.
  • Designed and deployed RAG-powered semantic search using LangChain and LlamaIndex on AWS SageMaker to support low-latency semantic retrieval and personalized content surfacing for media catalogs.
  • Automated ETL and feature engineering pipelines with AWS Glue, Pandas and PySpark, reducing preprocessing time by 35% and improving model training throughput for large-scale interaction data.
  • Introduced internal ML automation using code-generation tools to accelerate routine pipeline scripting and tooling, cutting engineering turnaround on recurring tasks and improving developer productivity.
  • Optimized production inference pipelines through MLflow experiment tracking, hyperparameter tuning, CI/CD integration and AWS monitoring, improving inference efficiency and deployment reliability by 22%.
Prudential Financial, USA
USA
AI/ML Engineer
Aug 2024 – April 2025
Delivered ML solutions for insurance analytics, underwriting and regulatory compliance; processed policy and claims data to support enterprise reporting and risk workflows.
Tech Stack: Scikit-learn, XGBoost, PySpark, Azure Synapse Analytics, Azure AI, LangChain, ChromaDB
  • Built and evaluated Scikit-learn and XGBoost models on policy and claims datasets to identify retention drivers and improve risk prediction, increasing model accuracy by 16%.
  • Processed high-volume structured and unstructured insurance records using SQL, PySpark and Azure Synapse Analytics to feed underwriting models and enterprise reporting.
  • Applied Azure AI Named Entity Recognition to extract financial and policy entities from documents, automating parts of document review and reducing manual extraction effort.
  • Designed a regulatory compliance RAG pipeline using LangChain, ChromaDB and Azure AI Search for enterprise document indexing and cross-referencing of policy language, cutting manual lookup time by 30%.
  • Implemented model monitoring and explainability using Azure Monitor and SHAP reports and integrated Azure DevOps CI/CD, reducing validation and deployment turnaround by 28%.
  • Accelerated pipeline scripting and unit-test coverage using GitHub Copilot, shortening development cycles for validation jobs and automation scripts.
Tata Consultancy Services (TCS), India
India
ML Engineer
Aug 2021 – Jul 2023
Delivered predictive analytics and data engineering solutions for enterprise clients; focused on forecasting, data consolidation, and ML deployment automation.
Tech Stack: Scikit-learn, XGBoost, AWS Glue, MLflow, Docker, PostgreSQL, MongoDB
  • Developed predictive analytics solutions using Scikit-learn and XGBoost to forecast customer behavior and operational trends, improving decision-making accuracy by 22% for enterprise reporting.
  • Consolidated large-scale transactional and operational datasets from PostgreSQL, MongoDB and data warehouse environments to support forecasting and KPI analysis pipelines.
  • Built classification and regression pipelines using Random Forests and SVM with feature engineering to strengthen customer segmentation and operational performance analysis.
  • Migrated legacy reporting into automated ETL pipelines using AWS Glue, Pandas and SQL, reducing manual reporting effort by 40% across analytics teams.
  • Configured MLflow, Docker and CI/CD pipelines to maintain model versioning and streamline deployments, improving release efficiency for enterprise ML applications.
  • Delivered interactive BI dashboards with Power BI and Tableau to monitor operational metrics and accelerate reporting turnaround by 30% for cross-functional stakeholders.

Projects

Enterprise RAG Knowledge Assistant
Tools Used: LangChain, LlamaIndex, Pinecone, PostgreSQL, TypeScript, Docker
  • Built a Retrieval-Augmented Generation platform using LangChain for LLM orchestration and LlamaIndex for document indexing to enable contextual retrieval across enterprise documentation.
  • Implemented semantic search and vector storage with Pinecone, used PostgreSQL for metadata and caching strategies to improve response relevance for internal knowledge queries.
  • Containerized LLM workflows and tracked experiments using Docker and MLflow to support scalable deployment and monitoring of AI-powered search services.
Real-Time Recommendation Engine
Tools Used: PyTorch, Neo4j, AWS Kinesis, PySpark, PostgreSQL
  • Designed a real-time recommendation engine using PyTorch with Neo4j graph modeling to personalize content ranking based on user interactions.
  • Orchestrated streaming pipelines with AWS Kinesis and PySpark to process high-volume engagement events and update recommendations dynamically.
  • Tuned collaborative filtering and ranking models with Optuna and Scikit-learn to improve click-through rate by 18% in prototype deployments.
MLOps Pipeline Automation
Tools Used: MLflow, Docker, Kubernetes, GCP Vertex AI
  • Engineered automated ML deployment workflows using MLflow, Docker and Kubernetes to streamline model versioning and container orchestration across clouds.
  • Provisioned training and inference environments on GCP Vertex AI and implemented monitoring to reduce model degradation incidents and improve stability.
  • Automated validation and CI/CD processes with Jenkins and pipeline scripts to accelerate reliable releases in multi-cloud environments.

Education

University of Massachusetts, Boston
Master of Science in Computer Science • Boston, MA • May 2025

Certifications

AWS Certified Machine Learning Engineer Associate — Amazon Web Services
Microsoft Certified: Azure AI Engineer Associate — Microsoft

Powered by Drivetube · Create your own profile at drivetube.ai

Everything on Drivetube

Six products, two of them free forever. Start wherever you are.

Read the docs →
  • Your free digital resume

    at drivetube.ai/in/your-name: one true standard resume with a Hiring Snapshot (visa status, expected salary, notice period, work preference, relocation), an ATS-ready PDF download and a single shareable link. Free forever; interview requests come from verified employers and your contact details stay masked until you accept.

    Documentation
  • Verified jobs, posted in the last 3 days

    verified openings crawled ATS-by-ATS from 100,000+ real company career pages across 35 ATS platforms. Shows the true posting date from the source ATS — not when a listing was indexed — and deletes every general listing 3 days after it was actually posted. No ghost jobs, no ad-sponsored listings, no staffing reposts, no account needed.

    Documentation
  • Job Hunt ProgramFrom $199.99

    We run your job hunt for you

    managed job hunting, a one-time purchase from $199.99. JobScout matches verified roles to your real experience band, Blend AI writes a uniquely tailored resume and cover letter for every application, and the Autofill extension fills the form — or Let Us Apply submits it for you.

    Documentation
  • Written by senior career writers

    human-written, ATS-optimised resumes by senior career writers, from ₹499.99 / $25.99. Available in every country, written to the destination country's own standard — a US resume, UK CV, German Lebenslauf and Indian resume are genuinely different documents. A paid service, separate from the free Drivetube Profile.

    Documentation
  • Community MembershipFrom $49.99/yr

    AI tools, gated filters and a $10,000+ library

    from $49.99/year (₹1,999/year in India), or $4.99/month in the iOS app. Unlocks the gated job-board filters — visa sponsorship, security clearance, workplace and application time — plus Job-Scout AI matching, Resume Report AI, Interview AI prep sheets, Recruiter Outreach AI sent from your own Gmail, Apply or Skip triage, a daily market feed and a $10,000+ library including 23 ATS-validated resume templates.

    Documentation
  • For employers — no job postings, no applications

    hiring with no job postings and no applications. Paste your real job description and AI matches it against candidates' true standard resumes, returning ranked candidates with a match %, matched and missing skills and written reasoning. Free tier included; employers pay, candidates never do.

    Documentation
  • Documentation

    every product explained in full, with feature-by-feature comparisons against the job boards, AI apply tools, resume services and hiring platforms people actually use.

  • Playbooks

    tactical job-search guides. Each post is one named tactic with the situation it applies to, the exact steps, a copy-paste message and honest failure modes. Free, no account.

The job board covers the United States, India, United Kingdom, Canada, Europe and Australia, and Resume Writing Services are available in every country.

Pricing is identical on the web, iOS and Android — no app-store markup. Monthly Community membership is purchased in the iOS app only; the website sells annual plans. The Android app has no in-app purchase, so Android members subscribe on drivetube.ai and then sign in to the app with full access. Membership follows your account, not your device.