This role is for one of the Weekday's clients
Salary range: Rs 2500000 - Rs 8000000 (ie INR 25-80 LPA)
Min Experience: 3 years
Location: Remote (India)
JobType: full-time
We are seeking exceptional full-stack ML engineers who can work independently, navigate complex systems, and contribute to building advanced environments used for training next-generation coding agents.
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)