ML Engineer

As a Machine Learning Engineer at Ailit, you’ll build the intelligence that powers AilitOS, our real-time energy orchestration engine.

You’ll develop predictive and optimization models that determine when, where, and how our battery-electric freight trains charge - balancing energy cost, route performance, and grid conditions in real time.

Your work will turn vast streams of operational and grid data into actionable decisions, shaping how the world’s first fully electric freight network operates.

• Design and implement ML models for predictive energy scheduling, grid price forecasting, and route optimization.
• Collaborate with software, energy, and systems engineers to integrate algorithms directly into real-time control systems.
• Build and maintain simulation environments to train and validate orchestration strategies.
• Use telemetry and historical data to continuously improve fleet-level efficiency and reliability.
• Work on optimization problems across multiple time horizons - from millisecond control to day-ahead energy planning.

• Have experience in applied machine learning, optimization, or control systems.
• Are comfortable working with energy, transport, or industrial data at scale.
• Understand reinforcement learning, simulation modeling, or time-series prediction.
• Enjoy collaborating closely with engineers working on both hardware and software.
• Are excited by the idea of building an AI that orchestrates megawatts, not ads.

Ailit is redefining freight rail through battery-electric power and AI-driven energy optimization.

We design locomotives, PowerWagons, and the orchestration software that keeps them moving - replacing diesel with intelligent, zero-emission energy systems.

We value engineers who think across disciplines and are motivated by large, physical-world problems.