Role Overview
We are looking for an experienced Senior Software Engineer or Technical Lead to design and build scalable machine learning platforms for training, optimizing, evaluating, packaging, and deploying large-scale AI models.
The role requires solid expertise in distributed systems, GPU computing, ML infrastructure, and production-grade software engineering. You will work closely with applied scientists, platform engineers, hardware teams, compiler teams, and infrastructure specialists.

Key Responsibilities
Design and develop distributed ML platform services and reusable libraries.
Build scalable training capabilities for large language and multimodal models.
Support data, tensor, pipeline, and model parallelism across multi-node GPU clusters.
Improve training throughput, GPU utilization, memory efficiency, communication performance, and fault recovery.
Define stable APIs and platform interfaces for model onboarding, training, evaluation, and deployment.
Integrate model optimization techniques such as quantization, pruning, knowledge distillation, and compression.




Build evaluation and artifact-management workflows to measure model quality and system performance.
Develop automated validation, CI/CD, regression testing, observability, and release pipelines.
Profile GPU workloads and resolve end-to-end system performance bottlenecks.
Build production-operational mechanisms, including metrics, alarms, dashboards, runbooks, and root-cause corrective actions.
Collaborate with model, compiler, runtime, hardware, security, infrastructure, and product teams.
Prepare technical designs, evaluate architecture trade-offs, and drive engineering decisions.
Mentor engineers and improve code reviews, design reviews, testing, and development practices.

Required Qualifications
Bachelor’s degree in Computer Science, Engineering, or a related technical field.
5+ years of professional software development experience.
Strong programming skills in C++, Java, C#, Python, or a simi

📌 Technical Lead – ML Platform (Bengaluru)
🏢 NYXSES India Private
📍 Bengaluru