Parshant Sharma

Machine Learning Engineer

India2 yrs 9 mos experience
AI EnabledAI ML Practitioner

Key Highlights

  • Expert in AI compilers and performance optimization.
  • Contributed to next-gen PyTorch compiler improvements.
  • Gold Medalist in Artificial Intelligence and Machine Learning.
Stackforce AI infers this person is a Machine Learning Engineer with expertise in AI compilers and deep learning technologies.

Contact

Skills

Core Skills

Machine LearningAi Compilers

Other Skills

PyTorchPerformance optimizationGraph transformationExport workflowsCompiler passesIR transformationsCompiler improvementsRuntime optimizationsDeep Learning compilersTensorFlowC++MLOpsLLVMMLIRLarge Language Models (LLM)

Experience

2 yrs 9 mos
Total Experience
2 yrs
Average Tenure
9 mos
Current Experience

Red hat

2 roles

Machine Learning Engineer

Promoted

Apr 2026Present · 1 mo

Associate Machine Learning Engineer

Aug 2025Apr 2026 · 8 mos

  • Contributing to core compiler and graph capture infrastructure within the PyTorch, with a focus on performance optimization, graph transformation, and export workflows
  • Working on TorchDynamo and torch.compile to improve graph capture, tracing stability, backend integrations, and execution efficiency.
  • Enhancing the Export API for stable graph serialization and deployment across diverse runtimes and accelerators.
  • Contributing to compiler passes, IR transformations, across the PyTorch stack.
  • Debugging and resolving edge cases in graph breaks, dynamic shapes, and operator lowering.
  • Additionally, contributing to the PyTorch Helion project, focusing on next-generation compiler/runtime improvements aimed at scalable, high-performance model execution.
PyTorchPerformance optimizationGraph transformationExport workflowsCompiler passesIR transformations+2

Krutrim

SDE

Aug 2023Aug 2025 · 2 yrs · Bengaluru · On-site

  • R&D on Deep Learning compilers for low power accelerators.
Deep Learning compilersTensorFlowC++Machine Learning

Intel corporation

Graduate Technical Intern

Aug 2022May 2023 · 9 mos · Bengaluru, Karnataka, India

Machine LearningMLOps

Education

CHANDIGARH UNIVERSITY

Master of Engineering - MEng — Artificial Intelligence and Machine Learning

Jan 2021Jan 2023

Punjab Technical University

Bachelor of Technology - BTech — Information Technology

Jul 2017Jun 2021

Stackforce found 100+ more professionals with Machine Learning & Ai Compilers

Explore similar profiles based on matching skills and experience