Machine Learning Engineer

Job location: Remote in India

About the role:

We are looking for a seasoned Machine Learning Engineer with deep expertise in deep learning to design and build production-grade models in the speech/audio and computer vision domains. The Machine Learning Engineer will own the end-to-end model development lifecycle - from dataset curation and architecture design through training, optimization, and deployment, and will work closely with software and product teams to ship low-latency, scalable AI features.

What you would be expected to do:

• Research, design, train, and productionize deep learning models for speech/audio (ASR, speaker diarization, audio classification, TTS, noise suppression) and computer vision (detection, segmentation, classification, video understanding) use cases.
• Architect training pipelines capable of handling large-scale datasets, manage data preprocessing, augmentation, and versioning workflows.
• Select and adapt state-of-the-art architectures (Transformers, CNNs, RNNs, diffusion models, etc.) and fine-tune or distill pre-trained models for production constraints.
• Optimize models for inference - quantization, pruning, knowledge distillation - targeting latency, throughput, and memory budgets on CPU/GPU/edge hardware.
• Work with software engineers to integrate models into production systems, design APIs and microservices for model serving.
• Define and track evaluation benchmarks, monitor model performance in production, and drive continuous improvement cycles.
• Stay current with research literature, evaluate and implement relevant SOTA techniques, contribute to internal technical forums.

You might be a strong candidate if you have/are:

• B.Tech / M.Tech / M.Sc. / PhD in Computer Science, Electrical Engineering, Signal Processing, or a related field.
• 4+ years of experience in machine learning/deep learning engineering with significant hands-on work in speech/audio and/or computer vision in production environments.
• PyTorch (primary) and TensorFlow/Keras, familiarity with JAX is a plus.
• Deep expertise in Wav2Vec 2.0, Whisper, ESPnet, SpeechBrain, torchaudio, librosa.
• Strong experience with speech/audio frameworks and toolkits: YOLO, DETR, ViT, EfficientNet, SAM, and standard CV pipelines (OpenCV, torchvision, Albumentations).
• Solid background in computer vision: TensorRT, ONNX Runtime, TorchScript, DeepSpeed, or Triton Inference Server.
• Hands-on experience with model inference optimization using Python and strong software engineering fundamentals: OOP, clean code, testing, CI/CD.
• Proficient in SQL for data extraction and pipeline logging.
• Experience with distributed training frameworks (DDP, DeepSpeed, FSDP) and large-scale data pipelines.
• Familiarity with cloud ML platforms (AWS SageMaker, GCP Vertex AI, Azure ML) and containerized deployment (Docker, Kubernetes, ECS).
• Good knowledge of system design principles for ML services: throughput, latency, scalability, fault tolerance.

What Sun King offers:

• Professional growth in a dynamic, rapidly expanding, high-social-impact industry
• An open-minded, collaborative culture made up of enthusiastic colleagues who are driven by the challenge of innovation towards profound impact on people and the planet.
• A truly multicultural experience: You will have the chance to work with and learn from people from different geographies, nationalities, and backgrounds.
• Structured, tailored learning and development programs that help you become a better leader, manager, and professional through the Sun King Center for Leadership.

We place great importance on sustaining a diverse, inclusive work environment:

We believe that innovation and understanding comes from diversity on every spectrum. We work to make sure every Sun King team member knows that they belong, knowing that sustaining an inclusive workplace requires conscious effort and is a continuous journey, not an end-state.

Sun King recruits, employs, trains, compensates and promotes people based on their experience, skills, effort, and results. We explicitly prohibit discrimination on the basis of race, religion, caste, national origin, color, gender, marital status, family structure, sexual orientation, HIV/AIDS status, or disability.

Recruitment disclaimer:

Sun King maintains high ethical standards in its recruitment processes, and invitations for interviews and job offers are based entirely on merit and qualified experience. Sun King will never request payment in any form, whether monetary, as favors, or in kind, at any stage of our recruitment or selection process.