About Zillwork
Zillwork is a Singapore-incorporated AI company building technology that creates economic opportunity at scale. We are pre-launch, founder-led, and operating out of Chennai and Coimbatore.
We are technically ambitious and globally minded. Our stack includes fine-tuned multilingual speech models, real-time geo-intelligent matching at scale, and a homegrown AI engine designed from first principles. We are building for markets and communities that existing technology has never seriously attempted to serve.
We are a smart, focused team. Every hire shapes the culture and the product. If you want your work to matter — and you are energised by hard problems over comfort — read on.
The Role
As Senior ML Engineer (RL and Forecasting), you will build two of the platform’s most technically complex systems: a PPO reinforcement learning policy for fair multi-objective matching, and a Temporal Fusion Transformer demand forecasting system that predicts platform activity 7 days ahead by geo-cell and category.
What You Will Build
– PPO policy using RLlib (Ray) — multi-objective reward: quality + fairness + satisfaction
– Temporal Fusion Transformer for geo-cell demand forecasting at city scale
– Kafka time-series integration for real-time demand signals
– ClickHouse aggregations for demand-supply heatmaps
– Proactive alert system based on predicted demand shifts
What We Are Looking For
– 5+ years ML engineering with production deployments
– Reinforcement Learning hands-on: PPO, DQN, or policy gradient in production
– RLlib (Ray) or Stable Baselines preferred
– Time-series forecasting: TFT, Prophet, DeepAR, or LSTM in production
– Apache Kafka for ML feature pipelines
– Python, PyTorch, Kubeflow
Bonus: Multi-objective RL, geo-cell demand forecasting, ClickHouse real-time analytics
Apply: hr@zillwork.com | Subject: Senior ML Engineer RL Forecasting Application
…