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 will 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.