Medra Logo

Medra

Research Engineer, Post-training

Posted 19 Hours Ago
Be an Early Applicant
In-Office
San Francisco, CA, USA
Entry level
In-Office
San Francisco, CA, USA
Entry level
Develops post-training recipes and data pipelines for AI models, creates evaluations, builds agentic systems with custom tools, and integrates reasoning capabilities into live scientific workflows. The role collaborates with scientists, robotics engineers, and operations teams to improve experimental design and assay development while shaping the technical direction of a new machine learning team.
The summary above was generated by AI
What We're Building:

If you want to shape the future of science, come build with us. Medra is building Physical AI Scientists: robotic systems that work hand-in-hand with leading biopharma partners to enable scientific breakthroughs faster than ever before.

🤖 Physical AI that can operate scientific instruments with human-level dexterity.

🧪 Scientific AI that can analyze results, reason about next steps, and close the loop autonomously.

We shipped our first production system over a year ago, recently raised a $52M Series A, and are opening one of the largest autonomous labs in the US. We're a small, ambitious team and you'd be joining early.

The Team:
  • We’re a team of passionate, mission-driven engineers from companies like Tesla, Amazon, SpaceX, and Neuralink. We’re collaborative and love moving fast, both with our product and on team trips skiing or go-karting!

  • As a team, we love nerding out about engineering and robotics — plus other topics like race cars or cooking. We like learning new things and then sharing our new knowledge with each other.

  • Our team is opinionated and straightforward. We don’t mind intense discussions about design tradeoffs. If we have arguments or miscommunication, we resolve conflicts quickly and empathetically.

In this role, you will:
  • Define post-training recipes for our AI models — from deciding which problem matters and how to measure it, to engineering large data collections, to running ML experiments, to integrating post-trained models into production workflows

  • Build and own the post-training data pipelines integrating both internal data and public data

  • Create meaningful and trustworthy evaluations that tell us whether our models are improving scientific protocols and assay development

  • Develop agentic systems with context management and custom tool calls to surface new scientific insights about experimental design in real lab environments

  • Work closely with scientists, robotics engineers, and operation teams to bring reasoning capabilities into live experimental loops for leading biopharma partners

  • Shape the engineering culture and technical direction of a new machine learning team that's redefining how life science R&D gets done

Let's talk if you have:

  • Practical experience building AI-driven workflows into the real world

  • Strong problem solving skills for debugging complex systems

  • A clear grasp of probability, statistics, and ML fundamentals

  • Ability to own the post-training stack end-to-end: data pipelines, harnesses, RL environments, and agentic evaluations, even when things are loosely defined

  • Proficiency in Python and familiarity with at least one deep learning framework (e.g., PyTorch, JAX)

  • Experience with LLMs, post-training, reinforcement learning, or agentic systems

HQ

Medra San Francisco, California, USA Office

San Francisco, CA, United States

Similar Jobs

10 Days Ago
In-Office
Palo Alto, CA, United States
609K-900K Annually
Senior level
609K-900K Annually
Senior level
Artificial Intelligence • Software • Generative AI
Own the post-training pipeline for AI roleplay models, transforming raw data into improved model behavior through supervised fine-tuning, reinforcement learning, reranking, data curation, and evaluation. The role involves closing the loop from evaluation signals to training runs, defining problems independently, and working onsite with a small, fast-moving engineering team.
Top Skills: Reinforcement Learning (Rl)RerankingSupervised Fine-Tuning (Sft)
13 Days Ago
In-Office
San Francisco, CA, USA
250K-500K Annually
Entry level
250K-500K Annually
Entry level
Artificial Intelligence • Software
Develop and implement frontier-model post-training methods, including reinforcement learning, verifiable rewards, evaluation, and data-centric optimization. Design rigorous experiments across datasets, rewards, environments, and training strategies; investigate model behavior and failure modes; build scalable data-generation and evaluation pipelines; and create benchmarks, rubrics, and scoring systems. Collaborate with researchers, engineers, AI teams, customers, and domain experts while contributing to open-source tools, technical reports, blog posts, and research papers.
Top Skills: APIsCloud InfrastructureDapoDatabasesDistributed SystemsGrpoMachine LearningReinforcement LearningRlvr
14 Days Ago
Remote or Hybrid
San Francisco, CA, USA
165K-310K Annually
Senior level
165K-310K Annually
Senior level
Artificial Intelligence • Machine Learning • Software
Design, optimize, and deploy large language model training and post-training pipelines. Improve model quality through fine-tuning, reinforcement learning, preference optimization, evaluation, and experimentation. Build PyTorch-based infrastructure, optimize distributed multi-GPU training, diagnose performance and convergence issues, and develop production-ready AI systems. Collaborate with researchers, infrastructure engineers, platform teams, and customers while contributing to open-source projects and reusable training capabilities.
Top Skills: CudaDeepspeedDistributed TrainingDpoFsdpGpu Performance OptimizationGrpoHugging Face TransformersLightning FabricMegatron-LmMixed PrecisionMulti-Gpu SystemsNvidia MoltPeftPpoPythonPyTorchReinforcement LearningReward ModelingRlhfSglangSupervised Fine-TuningTensorrtTransformer-Based Language ModelsTritonTrlVllm

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account