G

Senior Research Engineer – AI Agent Systems (Post-Training & RL)

goaly ai • United State
Visa Sponsorship
Apply Now
AI Summary

Join Goaly AI as a research-engineering leader to design and optimize AI agent systems, focusing on reinforcement learning, post-training experiments, and scalable model improvements. You’ll bridge research and engineering, improving infrastructure, data pipelines, and agentic capabilities like tool use, reasoning, and long-horizon task completion. Requires strong Python, deep learning, and RL expertise with hands-on experimental rigor.

Key Highlights
Own the experimental loop for transforming base models into capable agents (RL, tool use, coding, planning).
Design data pipelines, environments, and evaluations with reproducibility and reliability as priorities.
Collaborate cross-functionally to debug infrastructure, optimize training pipelines, and translate research into production.
Key Responsibilities
Design and execute post-training experiments for agentic capabilities (tool use, coding, reasoning, long-horizon tasks).
Build realistic RL environments, task harnesses, and evaluations with robust reward signals and telemetry.
Develop data pipelines for training/evaluation, ensuring contamination control, difficulty balancing, and reproducibility.
Analyze model behavior for failure modes (reward hacking, mode collapse, distribution gaps) and iterate on training recipes.
Improve research workflows by enhancing experiment configuration, rollout inspection, and automated comparison tools.
Partner with systems engineers to debug cross-layer issues in rollout inference, distributed training, and data movement.
Translate successful research into stable, production-ready pipelines and contribute to the team’s long-term roadmap.
Technical Skills Required
Python Reinforcement Learning Deep Learning
Benefits & Perks
Hybrid work (4+ days/week in Palo Alto)
Visa sponsorship (H-1B, OPT/CPT)
Complimentary meals and snacks
Nice to Have
Experience with RLHF, reinforcement fine-tuning, or preference optimization.
Familiarity with PyTorch/JAX and distributed ML frameworks (FSDP, Megatron, Ray, VeRL).
Record of influential research, open-source contributions, or production model launches.

Job Description


About Us

We’re building toward a world where every company can become its own AI lab.

Goaly is a stealth AI startup founded by ex-Meta MSL engineers and researchers. Our mission is to dramatically lower the cost, time, and talent barriers to building proprietary AI — and make each generation of models faster and cheaper to build than the last.

Backed by leading AI investors and endorsed by frontier AI researchers and builders, we’re looking for exceptional new grads who want to work on hard, foundational AI systems problems with outsized ownership from day one.

About The Role

You will own the experimental loop that turns a capable base model into a useful agent. You will design tasks and environments, prepare training and evaluation data, run reinforcement-learning and related post-training experiments, diagnose model behavior, and convert results into better recipes and production models.

This is a research-engineering role. The best candidates are equally comfortable forming hypotheses, writing high-quality code, operating training pipelines, and investigating why a model or metric moved. You will work closely with RL systems, training, inference, product, and domain experts; when infrastructure slows the science, you will help improve the infrastructure rather than treating it as someone else's problem.

What you'll do

  • Design and run post-training experiments for agentic capabilities, including tool use, coding, reasoning, planning, long-horizon task completion, and recovery from failure.
  • Prepare high-quality training and evaluation data: define task distributions, curate and filter examples, control contamination, balance difficulty, and build reproducible data-generation pipelines.
  • Build realistic RL environments and task harnesses with clear interfaces, reliable resets, isolated execution, useful telemetry, and reward signals that are hard to game.
  • Develop evaluations that measure both capability and reliability. Create regression suites, behavioral slices, error taxonomies, and dashboards that connect aggregate metrics to concrete model failures.
  • Iterate on training recipes, including supervised warm starts, sampling strategies, reward design, verifiers, curricula, optimization choices, and reinforcement fine-tuning methods.
  • Analyze trajectories and model behavior to find reward hacking, shortcut learning, mode collapse, distribution gaps, and other failure modes; turn those findings into targeted experiments.
  • Improve the research workflow through better experiment configuration, rollout inspection, reproducibility, checkpoint evaluation, and automated comparison of runs.
  • Partner with systems engineers to debug cross-layer problems in rollout inference, environment execution, distributed training, and data movement.
  • Translate successful research ideas into stable, repeatable pipelines and help set the team's longer-term post-training roadmap.

You may be a good fit if you have

  • Strong Python and software-engineering skills, including the ability to turn ambiguous research ideas into reliable experimental systems.
  • Hands-on experience training, fine-tuning, or evaluating modern language models, or an exceptional record in a closely related ML research area.
  • Solid understanding of deep learning and optimization, plus enough reinforcement-learning intuition to reason about policies, rewards, sampling, credit assignment, and evaluation bias.
  • Excellent experimental judgment: you define controls, inspect data, validate metrics, keep results reproducible, and distinguish a real improvement from noise or leakage.
  • Ability to debug across model behavior, data, code, and distributed infrastructure without losing sight of the user-facing capability being improved.
  • Clear written and verbal communication and a track record of productive collaboration across research and engineering.

Strong pluses

  • Experience with RLHF, reinforcement fine-tuning, preference optimization, reward or verifier modeling, or large-scale online sampling.
  • Experience building agent environments, secure sandboxes, coding benchmarks, tool-use tasks, or long-horizon evaluations.
  • Familiarity with PyTorch or JAX and distributed ML systems; experience with frameworks such as FSDP, Megatron, DeepSpeed, Ray, VeRL, or related stacks.
  • A record of influential research, open-source contributions, technically ambitious independent projects, or production model launches.

How we work

  • Mission first. We choose work for its impact on the mission and take responsibility for the outcome, not just our assigned tasks.
  • High agency. We identify what is missing, form a plan, and move without waiting for perfect clarity.
  • Speed with rigor. We ship, measure, and iterate quickly while protecting correctness, safety, and reliability.
  • Flexible scope. We cross team and technical boundaries when that is the fastest way to solve the real problem.
  • Low ego, high standards. We give direct feedback, change our minds when the evidence changes, and help the whole team win.
  • Continuous learning. The stack changes quickly; we are willing to learn unfamiliar systems, methods, and domains as the work demands.

Location, visa sponsorship & benefits

  • Hybrid in Palo Alto: 4+ days/week in office.
  • Visa sponsorship: H-1B and OPT/CPT support available, with immigration counsel.
  • Meals & perks: Complimentary lunch, dinner, snacks, and drinks.

A note on qualifications. We value exceptional ability over perfect keyword matches. If the work excites you and you can show strong technical ability, learning speed, or ownership, we encourage you to apply.

Equal opportunity

We are an equal opportunity employer. We consider qualified applicants without regard to race, color, religion, sex, sexual orientation, gender identity or expression, national origin, age, disability, veteran status, genetic information, or any other characteristic protected by applicable law. We provide reasonable accommodations for candidates who need them during the hiring process.


Similar Jobs

Explore other opportunities that match your interests

Visa Sponsorship Relocation Remote
Job Type Full-time
Experience Level Not Applicable

goaly ai

United State
Visa Sponsorship Relocation Remote
Job Type Full-time
Experience Level Entry level

BeaconFire Inc.

United State
Visa Sponsorship Relocation Remote
Job Type Full-time
Experience Level Not Applicable

Deloitte

United State

Subscribe our newsletter

New Things Will Always Update Regularly