Senior ML Engineer — Reinforcement Learning and Prediction

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: [Confidential Information] | Subject: Senior ML Engineer RL Forecasting Application