Artificial Intelligence AI Scientist

As an AI Engineer at our client's company in Pune, India, you will be responsible for designing, building, and scaling systems to power thousands of riders across India as part of the Last Mile 2.0 urban logistics project. Key Responsibilities: - Build and train forecasting models - Integrate model outputs into the real-time decision engine using Python APIs, Redis, and AWS - Create training/evaluation sets using simulation data - Optimize models for inference speed using ONNX/TorchScript and quantization techniques - Collaborate with the founders to shape the model roadmap and define metrics Qualifications: - Strong proficiency in Python along with PyTorch/TensorFlow - Experience with time-series models and data related to traffic, mobility, maps, and forecasting - Understanding of geospatial frameworks - Familiarity with data engineering concepts such as pipelines, ETL, and SQL/NoSQL - Ability to quickly deploy, debug, and optimize models for real-time serving Nice to have: - Experience with real-time infrastructure tools like Redis, OSRM, or similar - Background in last-mile logistics, ride-hailing, food and grocery delivery, or mobility data - Exposure to reinforcement learning or simulation-driven training Join our team for: - Early engineer role with significant ownership - Competitive salary package and meaningful equity - Opportunity to work on challenging real-world problems where ML decisions directly impact operational efficiency As an AI Engineer at our client's company in Pune, India, you will be responsible for designing, building, and scaling systems to power thousands of riders across India as part of the Last Mile 2.0 urban logistics project. Key Responsibilities: - Build and train forecasting models - Integrate model outputs into the real-time decision engine using Python APIs, Redis, and AWS - Create training/evaluation sets using simulation data - Optimize models for inference speed using ONNX/TorchScript and quantization techniques - Collaborate with the founders to shape the model roadmap and define metrics Qualifications: - Strong proficiency in Python along with PyTorch/TensorFlow - Experience with time-series models and data related to traffic, mobility, maps, and forecasting - Understanding of geospatial frameworks - Familiarity with data engineering concepts such as pipelines, ETL, and SQL/NoSQL - Ability to quickly deploy, debug, and optimize models for real-time serving Nice to have: - Experience with real-time infrastructure tools like Redis, OSRM, or similar - Background in last-mile logistics, ride-hailing, food and grocery delivery, or mobility data - Exposure to reinforcement learning or simulation-driven training Join our team for: - Early engineer role with significant ownership - Competitive salary package and meaningful equity - Opportunity to work on challenging real-world problems where ML decisions directly impact operational efficiency