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.
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
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
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
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
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.