Robotics ML Expert — AI Simulation & MuJoCo
alignerr
Job Description
What You'll Do
- Design, develop, and iterate on MuJoCo simulation environments for robotics research and AI training
- Implement and tune reinforcement learning algorithms (PPO, SAC, TD3, etc.) to train agents in simulated tasks
- Define reward functions, observation spaces, and action spaces that produce robust, transferable policies
- Debug and optimize physics simulations — contact models, actuator dynamics, and scene configurations
- Evaluate trained policies for stability, generalization, and sim-to-real transfer potential
- Document environment specifications, training procedures, and experimental results clearly and thoroughly
- Collaborate asynchronously with research teams to align simulation work with broader project goals
- Stay current with the latest advances in robot learning, simulation, and embodied AI
Who You Are
- Strong hands-on experience with MuJoCo (or MuJoCo via dm_control, Gymnasium/Gymnasium-Robotics, or similar wrappers)
- Solid understanding of reinforcement learning theory and practical training pipelines
- Proficient in Python and comfortable with ML frameworks such as PyTorch or JAX
- Experienced in defining and shaping reward functions for complex robotic tasks
- Familiar with robot kinematics, dynamics, and control fundamentals
- Able to read and write MJCF/XML model files and understand their physics implications
- Self-directed, detail-oriented, and comfortable working independently in an async environment
- Strong written communicator who can document technical work clearly
Nice to Have
- Experience with sim-to-real transfer techniques (domain randomization, system identification)
- Familiarity with other physics simulators — Isaac Gym, PyBullet, Drake, or Genesis
- Background in multi-agent environments or hierarchical RL
- Published research or open-source contributions in robotics, RL, or embodied AI
- Experience with imitation learning, model-based RL, or world models
- Graduate-level coursework or degree in robotics, ML, computer science, or a related field
Why Join Us
- Work on cutting-edge robotics and AI simulation projects alongside leading research labs
- Fully remote and flexible — work when and where it suits you
- Freelance autonomy with the structure of meaningful, milestone-driven work
- Directly influence how AI agents learn to interact with the physical world
- Engage with a global community of top-tier ML and robotics practitioners
- Potential for ongoing work and contract extension as new projects launch