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.
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.
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
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
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.
Research and develop large language models by designing experiments, curating data, optimizing transformer architectures, and training models on distributed infrastructure for production deployment.
Develop and validate radar-based precipitation algorithms and products using dual-polarization data, while analyzing radar variables, troubleshooting data quality issues, and collaborating on new platform capabilities.
Design, validate, and scale AI-powered research methods and products, while supporting team operations and building internal fluency in new research methods and AI tools.
Own computational and experimental protein-design work, improving models and closing the design-make-test loop through laboratory testing, kinetic analysis, and iterative design refinement.
Develop scientific benchmarks and ground-truth datasets to evaluate how AI agents perform diagnosis, treatment, research, and target discovery workflows for rare and progressive diseases.
