Skip to content
TensorStaxTensorStaxSan Francisco, CA

Research Engineer Intern, Evaluations

Designs evaluation frameworks and benchmarks to test AI agents' autonomy, reasoning, and reliability in data pipelines and warehouses. Requires experience in LLM benchmarking, reinforcement learning, Python, PyTorch/JAX, and data engineering tools.

Salary not listed
HybridML Engineering

About the role

What You’ll Do

  • Develop evaluation environments to test AI agents' ability to reason, plan, and act autonomously within mission-critical data pipelines.
  • Design benchmarks to assess model capabilities in failure detection, pipeline optimization, and agentic decision-making in data workflows.
  • Implement automated assessment frameworks for language model-based agents operating over data lakes and warehouses.
  • Work with synthetic and real-world datasets to create robust testing environments for AI-driven data automation.
  • Collaborate with research engineers to refine reward shaping strategies, guiding models toward more efficient and agentic behaviors in data-intensive tasks.

What We’re Looking For

  • Experience in language model research, with a focus on benchmarking LLMs in mission-critical domains.
  • Strong background in AI evaluation methodologies, reinforcement learning, and RLHF techniques.
  • Familiarity with benchmarking language models for structured and unstructured data tasks.
  • Proficiency in Python and experience with ML frameworks like PyTorch or JAX.
  • Hands-on experience with data lakes, warehouses, and data engineering tools (Snowflake, BigQuery, dbt, Spark, Kafka).
  • High agency—proactive, resourceful, and comfortable working in a fast-paced research environment with minimal supervision.
  • Attention to detail—ability to design rigorous, reproducible experiments and evaluations.

Bonus Points

  • Contributions to open-source AI benchmarks (e.g., SweBench, BIRD, SPIDER).
  • Contributions to open-source agentic frameworks.
  • Experience developing custom RL environments for AI evaluation.
  • Strong understanding of ETL, ELT, and data transformation pipelines.

Skills

PythonPyTorchJAXReinforcement LearningRLHFSnowflakeBigQuerydbtSparkKafka

Similar roles

ML Engineering jobs
Nuro

Software Engineer, Performance

NuroMountain View, CA

New grad software engineer optimizing performance of autonomous vehicle software. Profile, debug, and reduce latency of C++ systems running on x86, ARM, and GPUs while building tools for memory management and high-performance code.

153k – 175k/yr
On-siteEntry levelML Engineering
Airbnb

Machine Learning Engineer, Relevance and Personalization

AirbnbUnited States

Build and productionize cutting-edge ML models and ranking algorithms for Airbnb's search, recommendation, and personalization systems at scale. Requires PhD (new grad) or 2+ years applied ML experience with strong programming and ML systems skills.

166k – 195k/yr
Remote2+ YOEML Engineering
Abridge

Software Engineer

AbridgeSan Francisco, CA +1

Early-career Software Engineer building and iterating on agentic LLM systems, RAG pipelines, evaluation frameworks, and backend/frontend components for AI-powered clinical documentation at Abridge. Requires CS degree or equivalent plus hands-on GenAI experience from projects, internships or coursework.

157k – 184k/yr
On-siteEntry levelML Engineering
Mirage

ML Engineer, Generative Video

MirageNew York, NY

Build and scale video generation models at Mirage. Optimize training/inference for low-latency, real-time performance using PyTorch, CUDA, and distributed systems. Requires 2+ years industry experience in deep learning infrastructure.

175k – 275k/yr
On-site2+ YOEML Engineering
Traba

Software Engineer

TrabaNew York, NY

Build and own production AI agent systems (harnesses, evals, orchestration) on frontier LLMs for industrial supply chain workflows at Traba. Requires 1+ years shipping LLM/agent features to production, strong Python/TS skills, and high-agency in ambiguous customer environments.

140k – 200k/yr
On-site1+ YOEML Engineering