ML Engineer

This role is for one of the Weekday's clients

Salary range: Rs 2500000 - Rs 8000000 (ie INR 25-80 LPA)

Requirements

What We're Looking For

• Experience working in small, high-ownership environments (e.g., early-stage startup, founding engineer, CTO).
• Deep expertise in at least one core stack: TypeScript / Python / Go / Java — with a strong preference for backend experience and familiarity with TypeScript.
• Background as a product engineer, platform engineer, or ML engineer (not infrastructure-only or data-science-only profiles).
• Ability to understand, reason about, and solve difficult technical problems independently, while also effectively using tools like Cursor to accelerate development.
• Strong engineering fundamentals and comfort navigating large, unfamiliar codebases quickly.
  

What You Will Do

• Build reinforcement learning environments and sandboxes used by major AI labs to train advanced coding agents.
• Onboard to a new codebase every 1-1.5 months, develop a deep understanding of it, and construct RL tasks on top of it.
• Design engineering challenges, create mocks, write robust tests, and manipulate infrastructure components as needed to build realistic RL scenarios.
• Develop tools, utilities, and agent-support systems that make environment creation faster and more efficient for the entire engineering team.
• Work across full-stack components where required, while focusing heavily on backend logic and environment behavior.
  

Why You Should Join

• Join an extremely talented and deeply technical founding team with strong backgrounds in engineering, reinforcement learning, and startup building.
• Be part of a mission to build one of the most talent-dense engineering teams in India.
• Work closely with the founders on a daily basis in a high-ownership, high-trust environment.
• Opportunity to work on some of the most exciting problems in the AI ecosystem and contribute directly to the evolution of agentic systems.
• Competitive compensation and the chance to be part of a rapidly scaling, high-momentum company backed by top-tier global investors.
• In-person office being set up in Bangalore, with flexibility for exceptional remote candidates.
  

Skills

• PyTorch
• Training ML models
• Deploying ML models
• Backend fundamentals (TypeScript / Python / Go / Java)
• Full-stack engineering (as needed)