Job description

ABOUT US: Fireworks is the platform for specialized intelligence, enabling companies to build, train, and serve AI models tailored to their own data, workflows, and products. Founded by the team behind PyTorch and backed by AMD, Atreides, Benchmark Capital, Index Ventures, Lightspeed, NVIDIA, Sequoia Capital, and TCV, Fireworks powers production AI with hundreds of state-of-the-art open models across text, image, embedding, audio, and multimodal workloads. Today, Fireworks is a Series D company valued at $17.5 billion, bringing together an ambitious, collaborative team that's building the future of enterprise AI. THE ROLE: As a Member of Technical Staff on the Research team, you’ll push the boundaries of generative AI, advancing LLMs and multimodal systems through foundational research. Your work will enhance model efficiency, accuracy, and scalability, directly shaping our high-performance AI infrastructure. You'll collaborate with top experts in deep learning, distributed systems, and optimization to bring cutting-edge research into real-world applications. You'll also have the opportunity to shape how some of the world’s leading companies build and deploy AI through the models and tools you help create. KEY 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 MINIMUM QUALIFICATIONS: - Research background in Artificial Intelligence, Machine Learning, Physics, or 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 other similar languages PREFERRED QUALIFICATIONS: - PhD degree in Computer Science, Computational Physics, Mathematics, or a similar field - Research and engineering experience demonstrated via grants, fellowships, patents, internships, work experience, and/or coding competitions - Experience having first-authored publications at peer-reviewed conferences or journals - Experience working with ML frameworks such as PyTorch, TensorFlow, or Jax WHY FIREWORKS? - Solve Hard Problems: Tackle challenges at the forefront of AI infrastructure, from low-latency inference to scalable model serving. - Build What’s Next: Work with bleeding-edge technology that impacts how businesses and developers harness AI globally. - Ownership & Impact: Join a fast-growing, passionate team where your work directly shapes the future of AI—no bureaucracy, just results. - Learn from the Best: Collaborate with world-class engineers and AI researchers who thrive on curiosity and innovation. Fireworks AI is an equal-opportunity employer. We celebrate diversity and are committed to creating an inclusive environment for all innovators.

Requirements

Must-haves

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
Research background in Artificial Intelligence, Machine Learning, Physics, or 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 other similar languages

Nice-to-haves

PhD degree in Computer Science, Computational Physics, Mathematics, or a similar field
Research and engineering experience demonstrated via grants, fellowships, patents, internships, work experience, and/or coding competitions
Experience having first-authored publications at peer-reviewed conferences or journals
Experience working with ML frameworks such as PyTorch, TensorFlow, or Jax
WHY FIREWORKS?
Solve Hard Problems: Tackle challenges at the forefront of AI infrastructure, from low-latency inference to scalable model serving.
Build What’s Next: Work with bleeding-edge technology that impacts how businesses and developers harness AI globally.
Ownership & Impact: Join a fast-growing, passionate team where your work directly shapes the future of AI—no bureaucracy, just results.
Learn from the Best: Collaborate with world-class engineers and AI researchers who thrive on curiosity and innovation.
Fireworks AI is an equal-opportunity employer. We celebrate diversity and are committed to creating an inclusive environment for all innovators.

Application prep

Use the role’s language and application details while tailoring your materials.

Keywords to mirror

aimachine learningpythonml

Platform: ashby
Effort: quick