San Mateo, United States
You apply directly with the employer. We never charge applicants.
$200K–$300K/yr
Published by the employer
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:
This role is designed for researchers finishing their PhD who want their work to reach production, not just publication. 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 be paired with a senior researcher as a mentor and given a real research problem from day one. You'll collaborate with top experts in deep learning, distributed systems, and optimization, and you'll see the models you help create deployed by some of the world's leading companies — often within weeks rather than years.
We welcome candidates who have completed within the last 6 months, or expected completion by December 2026.
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
Work alongside engineering teams to transition research prototypes into production-grade systems
Analyze empirical results, identify performance bottlenecks, and iterate quickly to improve model quality
Bring ideas and emerging directions from your doctoral research into our internal research agenda
MINIMUM QUALIFICATIONS:
PhD completed within the last 6 months, or expected completion by December 2026, in Computer Science, Machine Learning, Computational Physics, Mathematics, Electrical Engineering, or a similar field
Research background in Artificial Intelligence, Machine Learning, Physics, or a related area, demonstrated through your dissertation, publications, or research projects
Experience solving analytical problems using analytic and quantitative approaches
Experience communicating research to audiences with different backgrounds
Proficiency coding in Python, C/C++, or other similar languages
Hands-on experience with ML frameworks such as PyTorch, JAX, or TensorFlow
PREFERRED QUALIFICATIONS:
First-authored publications at top-tier peer-reviewed conferences or journals (NeurIPS, ICML, ICLR, MLSys, or similar)
Industry research internships, or open-source contributions to ML systems, model training, or inference projects
Research or engineering experience demonstrated via grants, fellowships, patents, or coding competitions
Experience with large-scale model training, distributed systems, GPU kernel development, or inference optimization
Experience taking research code beyond the prototype stage
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.