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 building modular battery systems for freight rail, enabling operators to reduce diesel consumption without changing train operations. We integrate batteries with our own power electronics, control software, and rail interfaces, assembling and operating the AI optimized energy systems for deployment in North America and other heavy-haul markets.

We are building the company with a strong focus on disciplined execution, technical depth, and long-term impact in critical industrial infrastructure.