Fireworksai

<h2>About Us:</h2>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.<h2>THE ROLE</h2>We are seeking the first Data Analytics Engineer at Fireworks who will support our product management team.You will define canonical metrics that measure user behavior, product performance, system performance, and financial measures. You’ll ensure the metrics accurately measure what the business needs to know and that they are socialized and understood across teams and senior leaders. You will extend and improve our data model, write pipelines to compute metrics, and partner with the data platform team to put them into production following rigorous engineering standards.You’re a good fit for this role if you enjoy operating highly autonomously, if your working style stresses both efficiency and correctness, and if you love bringing structure to complex and messy questions.<h2>WHAT WE'RE LOOKING FOR</h2><h3>Required</h3><ul><li>5+ years in data science, product analytics, analytics engineering, or a combination.</li><li>Experience defining, analyzing, and aligning business users on shared metrics.</li><li>Experience building or working on production data pipelines, hands-on.</li><li>Proficiency in SQL, Python, or similar languages.</li><li>Excellent communication with senior level stakeholders, technical and non-technical.</li><li>Demonstrated autonomy in an environment with open ended problems.</li><li>Availability to work in office in San Mateo 3x days per week.</li></ul><h3>Preferred</h3><ul><li>Experience analyzing performance of self-service developer-facing products.</li><li>dbt, Coalesce, or a similar transformation framework in production — models, tests, docs, shared repo</li><li>Background in AI infrastructure including inference, GPU-hour metering, attributing revenue to mixed model use, etc.</li></ul><h2>Why Fireworks?</h2><ul><li>Solve Hard Problems: Tackle challenges at the forefront of AI infrastructure, from low-latency inference to scalable model serving.</li><li>Build What’s Next: Work with bleeding-edge technology that impacts how businesses and developers harness AI globally.</li><li>Ownership &amp; Impact: Join a fast-growing, passionate team where your work directly shapes the future of AI—no bureaucracy, just results.</li><li>Learn from the Best: Collaborate with world-class engineers and AI researchers who thrive on curiosity and innovation.</li></ul>Fireworks AI is an equal-opportunity employer. We celebrate diversity and are committed to creating an inclusive environment for all innovators.

Data Analytics Engineer

<h2>About Us:</h2>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.<h2>The Role:</h2>As an Applied Machine Learning Engineer, you will serve as a vital bridge between cutting-edge AI research and practical, real-world applications. Your work will focus on developing, fine-tuning, and operationalizing machine learning models that drive business value and enhance user experiences. This is a hands-on engineering role that combines deep technical expertise with a strong customer focus to deliver scalable AI solutions.<h2>Key Responsibilities:</h2><ul><li>Customer Success: Collaborate directly with the GTM team (Account Executives and Solutions Architects) to ensure smooth integration and successful deployment of ML solutions.</li><li>Demo / Proof of Concept (PoC): Build and present compelling PoCs that demonstrate the capabilities of our AI technology.</li><li>Application Build: Design, develop, and deploy end-to-end AI-powered applications tailored to customer needs.</li><li>Platform Features / Bug Fixes: Contribute to the internal ML platform, including adding features and resolving issues.</li><li>New Model Enablements: Integrate and enable new machine learning models into the existing platform or client environments.</li><li>Performance Optimizations: Improve system performance, efficiency, and scalability of deployed models and applications.</li><li>Partnership Enablement: Work closely with partners to enable joint AI solutions and ensure seamless collaboration.</li></ul><h2>Minimum Qualifications:</h2><ul><li>Bachelor’s degree in Computer Science, Engineering, or a related technical field.</li><li>5+ years of experience in a software engineering role, with a strong preference for customer-facing roles.</li><li>Robust coding skills required, preferably with proficiency in Python.</li><li>Demonstrated ability to lead and execute complex technical projects with a focus on customer success.</li><li>Strong interpersonal and communication skills; ability to thrive in dynamic, cross-functional teams.</li></ul><h2>Preferred Qualifications:</h2><ul><li>Master’s degree in Computer Science, Engineering, or a related technical field.</li><li>Experience working in a startup or fast-paced environment.</li><li>Hands-on experience fine-tuning machine learning models, including supervised fine-tuning (SFT) and reinforcement learning from human feedback (RLHF or RFT).</li><li>Solid understanding of generative AI, machine learning principles, and enterprise infrastructure.</li></ul><h2>Why Fireworks?</h2><ul><li>Solve Hard Problems: Tackle challenges at the forefront of AI infrastructure, from low-latency inference to scalable model serving.</li><li>Build What’s Next: Work with bleeding-edge technology that impacts how businesses and developers harness AI globally.</li><li>Ownership &amp; Impact: Join a fast-growing, passionate team where your work directly shapes the future of AI—no bureaucracy, just results.</li><li>Learn from the Best: Collaborate with world-class engineers and AI researchers who thrive on curiosity and innovation.</li></ul>Fireworks AI is an equal-opportunity employer. We celebrate diversity and are committed to creating an inclusive environment for all innovators.

Member of Technical Staff, Applied Machine Learning Engineer

<h2>About Us:</h2>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.<h2>The Role: </h2>We're looking for a Software Engineer focused on Performance Optimization to help push the boundaries of speed and efficiency across our AI infrastructure. In this role, you'll take ownership of optimizing performance at every layer of the stack—from low-level GPU kernels to large-scale distributed systems. A key focus will be maximizing the performance of our most demanding workloads, including large language models (LLMs), vision-language models (VLMs), and next-generation video models.You’ll work closely with teams across research, infrastructure, and systems to identify performance bottlenecks, implement cutting-edge optimizations, and scale our AI systems to meet the demands of real-world production use cases. Your work will directly impact the speed, scalability, and cost-effectiveness of some of the most advanced generative AI models in the world.<h2>Key Responsibilities:</h2><ul><li>Optimize system and GPU performance for high-throughput AI workloads across training and inference</li><li>Analyze and improve latency, throughput, memory usage, and compute efficiency</li><li>Profile system performance to detect and resolve GPU- and kernel-level bottlenecks</li><li>Implement low-level optimizations using CUDA, Triton, and other performance tooling</li><li>Drive improvements in execution speed and resource utilization for large-scale model workloads (LLMs, VLMs, and video models)</li><li>Collaborate with ML researchers to co-design and tune model architectures for hardware efficiency</li><li>Improve support for mixed precision, quantization, and model graph optimization</li><li>Build and maintain performance benchmarking and monitoring infrastructure</li><li>Scale inference and training systems across multi-GPU, multi-node environments</li><li>Evaluate and integrate optimizations for emerging hardware accelerators and specialized runtimes</li></ul><h2>Minimum Qualifications:</h2><ul><li>Bachelor’s degree in Computer Science, Computer Engineering, Electrical Engineering, or equivalent practical experience</li><li>5+ years of experience working on performance optimization or high-performance computing systems</li><li>Proficiency in CUDA or ROCm and experience with GPU profiling tools (e.g., Nsight, nvprof, CUPTI)</li><li>Familiarity with PyTorch and performance-critical model execution</li><li>Experience with distributed system debugging and optimization in multi-GPU environments</li><li>Deep understanding of GPU architecture, parallel programming models, and compute kernels</li></ul><h2>Preferred Qualifications:</h2><ul><li>Master’s or PhD in Computer Science, Electrical Engineering, or a related field</li><li>Experience optimizing large models for training and inference (LLMs, VLMs, or video models)</li><li>Knowledge of compiler stacks or ML compilers (e.g., torch.compile, Triton, XLA)</li><li>Contributions to open-source ML or HPC infrastructure</li><li>Familiarity with cloud-scale AI infrastructure and orchestration tools (e.g., Kubernetes)</li><li>Background in ML systems engineering or hardware-aware model design</li></ul><h2>Example projects:</h2><ul><li>Implement fully asynchronous low-latency sampling for large language models integrated with structured outputs</li><li>Implement GPU kernels for the new low-precision scheme and run experiments to find optimal speed-quality tradeoff</li><li>Build a distributed router with a custom load-balancing algorithm to optimize LLM cache efficiency</li><li>Define metrics and build harness for finding optimal performance configuration (e.g. sharding, precision) for a given class of model</li><li>Determine and implement in PyTorch an optimal sharding scheme for a novel attention variant</li><li>Optimize communication patterns in RDMA networks (Infiniband, RoCE)</li><li>Debug numerical instabilities for a given model for a small portion of requests when deployed at scale</li></ul><h2>Why Fireworks?</h2><ul><li>Solve Hard Problems: Tackle challenges at the forefront of AI infrastructure, from low-latency inference to scalable model serving.</li><li>Build What’s Next: Work with bleeding-edge technology that impacts how businesses and developers harness AI globally.</li><li>Ownership &amp; Impact: Join a fast-growing, passionate team where your work directly shapes the future of AI—no bureaucracy, just results.</li><li>Learn from the Best: Collaborate with world-class engineers and AI researchers who thrive on curiosity and innovation.</li></ul>Fireworks AI is an equal-opportunity employer. We celebrate diversity and are committed to creating an inclusive environment for all innovators.

Member of Technical Staff, Performance Optimization

<h2>About Us:</h2>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.<h2>The Role:</h2>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.<h3>Key Responsibilities</h3><ul><li>Conduct foundational research to advance the capabilities, efficiency, and reliability of LLMs and multimodal systems</li><li>Design, implement, and evaluate novel model architectures, training methods, and optimization techniques</li><li>Collaborate with engineering teams to transition research prototypes into production-grade systems</li><li>Analyze empirical results, identify performance bottlenecks, and iterate quickly to improve model quality</li><li>Contribute to internal research strategy by identifying high-impact opportunities and emerging trends in AI</li></ul><h2>Minimum Qualifications:</h2><ul><li>Research background in Artificial Intelligence, Machine Learning, Physics, or similar field</li><li>Experience solving analytical problems using analytic and quantitative approaches</li><li>Experience communicating research to audiences with different backgrounds</li><li>Experience coding in C/C++, Python, or other similar languages</li></ul><h2>Preferred Qualifications:</h2><ul><li>PhD degree in Computer Science, Computational Physics, Mathematics, or a similar field</li><li>Research and engineering experience demonstrated via grants, fellowships, patents, internships, work experience, and/or coding competitions</li><li>Experience having first-authored publications at peer-reviewed conferences or journals</li><li>Experience working with ML frameworks such as PyTorch, TensorFlow, or Jax</li></ul><h2>Why Fireworks?</h2><ul><li>Solve Hard Problems: Tackle challenges at the forefront of AI infrastructure, from low-latency inference to scalable model serving.</li><li>Build What’s Next: Work with bleeding-edge technology that impacts how businesses and developers harness AI globally.</li><li>Ownership &amp; Impact: Join a fast-growing, passionate team where your work directly shapes the future of AI—no bureaucracy, just results.</li><li>Learn from the Best: Collaborate with world-class engineers and AI researchers who thrive on curiosity and innovation.</li></ul>Fireworks AI is an equal-opportunity employer. We celebrate diversity and are committed to creating an inclusive environment for all innovators.

Fireworksai

Open jobs at Fireworksai

Data Analytics Engineer

Member of Technical Staff, Applied Machine Learning Engineer

Member of Technical Staff, Performance Optimization

Member of Technical Staff, Research

Working for Fireworksai?

Featured companies