Skip to content
Fireworks AIFireworks AISan Mateo, CA

Member of Technical Staff, Research

Conducts foundational research on LLMs and multimodal systems, developing architectures, training methods, and optimization techniques and helping transition prototypes into production. The role requires an AI/ML research background, analytical problem-solving, programming experience, and strong research communication.

200k – 350k/yr
On-siteAI Research

About the role

Responsibilities

  • Conduct foundational research to advance the capabilities, efficiency, and reliability of LLMs and multimodal systems.
  • Design, implement, and evaluate novel model architectures, training methods, and optimization techniques.
  • Collaborate with engineering teams to transition research prototypes into production-grade systems.
  • Analyze empirical results, identify performance bottlenecks, and iterate quickly to improve model quality.
  • Contribute to internal research strategy by identifying high-impact opportunities and emerging trends in AI.

Requirements

  • Research background in artificial intelligence, machine learning, physics, or a similar field.
  • Experience solving analytical problems using analytic and quantitative approaches.
  • Experience communicating research to audiences with different backgrounds.
  • Experience coding in C/C++, Python, or similar languages.

Nice-to-Haves

  • PhD in computer science, computational physics, mathematics, or a similar field.
  • Research and engineering experience demonstrated through grants, fellowships, patents, internships, work experience, and/or coding competitions.
  • First-author publications at peer-reviewed conferences or journals.
  • Experience with machine learning frameworks such as PyTorch, TensorFlow, or JAX.

Benefits

  • Work on challenging AI infrastructure problems, including low-latency inference and scalable model serving.
  • Build with cutting-edge technology affecting how businesses and developers use AI.
  • Significant ownership and impact in a fast-growing team.
  • Collaboration with experienced engineers and AI researchers.

Skills

artificial intelligenceMachine LearningLLMsmultimodal systemsc/c++PythonPyTorchTensorFlowJAXDeep LearningDistributed Systemsoptimizationmodel architecturesModel Trainingscalable model serving

Similar roles

AI Research jobs
Astera

Senior/Staff AI Scientist

AsteraNew York, NY +1

Senior/Staff AI Research Scientist building post-training RL feedback loops and generative models that align frontier biological AI to high-throughput experimental measurements of folding, binding, and function. Requires PhD (or equivalent), hands-on training of models from scratch, and experience with diffusion/transformers/RL.

200k – 300k/yrHybrid7+ YOEAI Research
Polymath

Member of Technical Staff - Research

PolymathSan Francisco, CA

Conducts applied research on long-horizon autonomous AI agents, focusing on evaluation, post-training, environment design, and benchmarks to improve frontier models. Builds simulations, runs experiments, ships production code, and publishes findings.

200k – 350k/yrOn-siteAI Research
Webflow

Senior Staff Machine Learning Scientist, Assets

WebflowUnited States

Leads research in computer vision, multimodal understanding, and visual generation. Develops novel models and methodologies, translates research to production, and mentors teams. Requires PhD preferred, 8+ years experience, and expertise in PyTorch, TensorFlow, transformers.

194k – 285k/yrRemote8+ YOEAI Research
Nuro

Senior/Staff Machine Learning Research Scientist: Generative Modeling for Planning

NuroMountain View, CA

Develop and scale generative models like diffusion and flow-matching for autonomous driving plan generation. Collaborate across teams to productize models for real-world deployment, requiring PhD/MSc + 3+ years in generative modeling and strong Python/C++ skills.

194k – 352k/yrOn-siteAI Research
Databricks

Staff Research Engineer, Data Agents

DatabricksSan Francisco, CA

Develop post-training recipes and build enterprise data agents for autonomous planning, code generation, and multi-step workflows. Requires 2+ years applied research experience shipping prototypes, plus expertise in LLMs, agents, and RL.

190k – 270k/yrOn-site2+ YOEAI Research