P

Senior Machine Learning Engineer (Reinforcement Learning Environments)

preference model • United States
Visa Sponsorship Relocation
Apply Now

Design and build reinforcement learning (RL) environments for training advanced AI models. Requires 5+ years of ML experience, expertise in LLMs/transformers, and proficiency in Python/PyTorch or JAX. Focus on research-driven engineering to improve AI capabilities and alignment.

Key Highlights
Research-driven role focused on building RL environments for superintelligent AI models
Full ownership of projects with autonomy in design and implementation
Collaboration with top ML engineers and researchers in a high-impact startup environment
Key Responsibilities
Design and build RL environments and reward functions for frontier AI models
Conduct experiments and evaluations for model training and alignment
Collaborate with researchers and engineers to improve ML infrastructure
Technical Skills Required
Python PyTorch/JAX Reinforcement Learning (RL) Environments
Benefits & Perks
Competitive cash and equity compensation (>90th percentile)
Health, vision, dental, and 401K benefits
Visa sponsorship & relocation support
Lunch and weekly snack orders provided onsite
Nice to Have
Expertise in transformer internals and modern LLM training/inference
Publications or open-source contributions in ML research
Experience with kernel development (CUDA, Triton, Pallas)

Job Description

About Us

Preference Model is a superintelligence data research company. We build RL environments for training capable, better-aligned superintelligences.

Almost every aspect of how a model behaves is shaped by the reward signals it's trained on. The biggest problem with AIs today is that they don't always do what we intend. Sometimes that's because they aren't capable enough, and sometimes it's because they aren't aligned. Both problems come down to the training objectives.

Want the full job description? Read the complete details on LinkedIn, the original posting.
Continue on LinkedIn

This is a short excerpt. All rights to the full description belong to its original publisher.

See Jaabz jobs first on Google 1 tap · free · in AI Overviews Jaabz is on your Google Manage Preferred Sources
Keep exploring

Similar Jobs

Explore other opportunities that match your interests

Career Resources & Guides

Latest guides related to this job: Visas, Work Permits & Legal Insights, Relocation Guides.

Browse all articles

Subscribe to our newsletter

New visa sponsorship, relocation and remote tech jobs, delivered to your inbox.