Machine Learning Engineers, ML Systems & Inference Experts.

We are seeking experienced Machine Learning Engineers, ML Researchers, AI Engineers, ML Systems Engineers, Inference Engineers, and Performance Optimization Specialists to contribute to a technically challenging AI training project.

What You’ll Do

• Develop and validate machine learning models and components
• Build reproducible training and inference workflows
• Implement model architectures, tensor operations, data pipelines, tokenization, batching, and generation systems
• Review and debug AI-generated machine learning code
• Diagnose numerical instability, incorrect tensor behavior, memory issues, and performance regressions
• Optimize inference and training for latency, throughput, memory consumption, and hardware utilization
• Compare multiple model implementations for correctness, efficiency, and reproducibility
• Build benchmarks and objective evaluation criteria
• Develop tests that verify model behavior and performance
• Debug distributed and system-level ML failures




• Document technical decisions, engineering tradeoffs, assumptions, and limitations
• Create reference implementations and technically rigorous solutions

Who Can Apply

Relevant backgrounds include:

Machine Learning Engineers, Senior Machine Learning Engineers, Staff Machine Learning Engineers, Principal Machine Learning Engineers, Lead Machine Learning Engineers, Applied Machine Learning Engineers, and ML Software Engineers.

We also welcome:

AI Engineers, Senior AI Engineers, Artificial Intelligence Engineers, Applied AI Engineers, Generative AI Engineers, Deep Learning Engineers, Neural Network Engineers, and AI Systems Engineers.

Research backgrounds may include:

Machine Learning Researchers, Research Engineers, AI Research Engineers, Research Scientists, Applied Scientists, Deep Learning Researchers, NLP Researchers, Computer Vision Researchers, Reinforcement Learning Researchers, and Foundation Model Researchers.

ML sys

📌 ML Engineer | Remote (India)
🏢 The Ai Training
📍 India