ML Infrastructure Engineer

Byteridge; Bengaluru; On-site; INR 1,200,000 - 1,800,000; Full time

Job summary

Byteridge is looking for a Rapid Prototyping Engineer specializing in AI Infrastructure and Optimization in Bengaluru. This role requires expertise in deploying and optimizing large language models using AWS infrastructure and GPU technology.

The ideal candidate will have extensive experience in machine learning infrastructure and GPU optimization techniques, including hands-on work with training pipelines and performance benchmarks. A strong analytical skill set, coupled with excellent communication abilities, is essential to thrive in a fast-paced environment.

Qualifications

• 5+ years of experience in machine learning infrastructure, model deployment, or GPU computing.
• Deep understanding of large language model architectures and optimization techniques.
• Experience with multi-GPU/multi-node orchestration.

Responsibilities

• Lead end-to-end deployments of large language models on AWS infrastructure.
• Design and implement training and inference pipelines using Amazon SageMaker.
• Optimize model performance through GPU-level tuning.

Skills

Machine learning infrastructure

GPU optimization

Production ML systems

Python programming

AWS core services

Education

Bachelor's degree in Computer Science, Engineering, or equivalent

Tools

PyTorch

TensorFlow

JAX

Amazon SageMaker AI

Job description

Byteridge is seeking a Rapid Prototyping Engineer specializing in AI Infrastructure and Optimization to work with our most strategic customers on deploying, fine‑tuning, and optimizing large language models at scale. You will be at the forefront of Byteridge's AI infrastructure capabilities, helping customers unlock the full potential of foundation models through expert‑level deployment on GPU infrastructure. This highly technical role requires deep expertise in machine learning infrastructure, GPU optimization, and production ML systems, combined with the ability to translate complex technical concepts into customer success.

Core Responsibilities

• Lead end‑to‑end deployments of large language models on AWS infrastructure for strategic customers.
• Design and implement training, fine‑tuning, and inference pipelines using Amazon SageMaker AI.
• Optimize model performance through GPU‑level tuning, kernel optimization, and infrastructure configuration.
• Deploy models on diverse GPU architectures, including NVIDIA and AWS custom silicon (Trainium, Inferentia).

Infrastructure Architecture And Performance

• Architect scalable ML infrastructure using SageMaker AI Inference, HyperPod, and distributed training frameworks.
• Implement CUDA‑level optimizations and custom kernels for improved model performance.
• Design storage and networking architectures optimized for high‑throughput ML workloads.
• Troubleshoot and resolve complex performance bottlenecks at the GPU driver and kernel level.

Customer Engagement And Technical Leadership

• Partner with AWS AI Specialist Solution Architects and customer ML teams to understand model requirements and deployment constraints.
• Provide technical guidance on model selection, fine‑tuning strategies, and production best practices.
• Conduct performance benchmarking and cost optimization analysis for ML workloads.
• Share field insights with AWS product teams to influence infrastructure and service roadmaps.

Requirements

• Bachelor's degree in Computer Science, Engineering, or equivalent practical experience (Master’s or PhD preferred).
• 5+ years of experience in machine learning infrastructure, model deployment, or GPU computing.
• Strong programming skills in Python and experience with ML frameworks (PyTorch, TensorFlow, JAX).
• Deep understanding of LLM architectures, training methodologies, and inference optimization.

Technical Expertise (High‑Level Alignment)

• Hands‑on experience training, fine‑tuning, or deploying large language models in production.
• Proficiency with GPU programming, CUDA, and kernel‑level optimization techniques.
• Experience with distributed training frameworks and multi‑GPU/multi‑node orchestration.
• Strong knowledge of AWS core services: EC2 (GPU instances), S3, EFS, VPC, and networking.

Preferred Experience

• Direct experience with Amazon SageMaker AI (Training, Inference, HyperPod) or equivalent ML platforms.
• Understanding of GPU architectures (NVIDIA A100, H100) and AWS custom silicon (Trainium, Inferentia).
• Experience with model compression techniques (quantization, pruning, distillation).
• Knowledge of MLOps practices, model monitoring, and production ML system design.
• Background in high‑performance computing, distributed systems, or systems programming.

Essential Attributes

• Ability to dive deep into technical problems and debug complex infrastructure issues.
• Strong analytical skills with a data‑driven approach to optimization.
• Excellent communication skills to explain complex technical concepts to diverse audiences.
• Comfortable working in ambiguous, fast‑paced environments with evolving requirements.
• Ownership mindset with the ability to drive projects from architecture to production.