GPU Machine Learning Intern / Junior Engineer

About the Role

early-career ML engineer

Key Responsibilities

• Model Development & Training

• Assist in training and fine-tuning diffusion and transformer models using

  PyTorch

   (and occasionally TensorFlow or JAX).
• Experiment with

  mixed precision

   (FP16/BF16) and basic

  quantization

   for performance gains.
• Run and monitor GPU jobs, log results, and document experiments.

• Model Serving & Integration

• Support deployment of models with

  Triton Inference Server, vLLM, or TensorRT-LLM

   under the guidance of senior engineers.
• Help connect model endpoints to our

  front-end APIs

   (REST/gRPC/WebSockets) for real-time user experiences.

• GPU Systems & Performance

• Learn to work with

  CUDA kernels

   and GPU profiling tools (Nsight, nvprof) to identify bottlenecks.
• Assist in basic

  multi-GPU scheduling

   and understand concepts like NCCL, MIG partitioning, and NVLink.

• Collaboration & Documentation

• Work closely with senior ML engineers, front-end developers, and product designers.
• Maintain experiment logs, code comments, and internal documentation for reproducibility.

• Education:

• B.Tech/M.Tech (Computer Science, AI/ML, Data Science, Electrical, or related) – recent graduates welcome.

• Technical Skills:

• Solid Python programming and familiarity with

  PyTorch

  .
• Basic understanding of

  deep learning concepts

  , GPU computing, and Linux environment.
• Exposure to

  CUDA

   or willingness to learn.
• Familiarity with Git and basic DevOps practices.

• Mindset:

• Eagerness to learn GPU optimization and large-scale ML deployment.
• Strong problem-solving skills and attention to detail.
• Ability to work full-time on-site in Gurgaon Sector 49.
• Coursework or hobby projects in

  computer vision, diffusion models, or LLMs

  .
• Experience with

  ONNX Runtime

  ,

  OpenVINO

  , or

  TensorRT

  .
• Knowledge of cloud GPU platforms (AWS/GCP/Azure) or Kubernetes basics.

What We Offer

• Hands-on mentorship

   from senior engineers building cutting-edge generative AI.
• Access to

  state-of-the-art GPU infrastructure

  .
• Opportunity to grow into a full-time ML Systems Engineer role at MatchBest Group.

Job Type: