Member of Technical Staff
full-time
lead
Posted 2 weeks ago
Apply Now
Stand out: build a proof-of-work pitch →
Free GitHub-based preview. Direct apply stays one click away.
Get weekly job alerts like this →Hiring for this role?
About this role
About Us:
At Fireworks, we’re building the future of generative AI infrastructure. Our platform delivers the highest-quality models with the fastest and most scalable inference in the industry. We’ve been independently benchmarked as the leader in LLM inference speed and are driving cutting-edge innovation through projects like our own function calling and multimodal models. Fireworks is a Series C company valued at $4 billion and backed by top investors including Benchmark, Sequoia, Lightspeed, Index, and Evantic. We’re an ambitious, collaborative team of builders, founded by veterans of Meta PyTorch and Google Vertex AI.
The Role:
As a Training Infrastructure Engineer, you'll design, develop, and maintain large-scale backend and cloud-native infrastructure to support distributed machine learning training, inference, and data processing pipelines for our generative AI platform. You'll architect scalable, resilient backend infrastructure, lead technical design discussions, mentor engineers, and establish best practices for large-scale machine learning systems.
Key Responsibilities:
Architect and build scalable, resilient backend infrastructure to support distributed training, inference, and data processing pipelines
Lead technical design discussions, mentor engineers, and establish best practices for large-scale machine learning systems
Design and implement core backend services with a focus on efficiency and low latency
Drive infrastructure optimization initiatives for compute cost, storage lifecycle management, and network performance
Collaborate with machine learning, DevOps, and product teams to translate research and product requirements into robust infrastructure solutions
Evaluate and integrate cloud-native and open-source technologies such as Kubernetes, Ray, Kubeflow, and MLFlow to enhance platform reliability
Own end-to-end systems from design to deployment, emphasizing reliability, fault tolerance, and operational excellence
Minimum Qualifications:
Bachelor's degree or equivalent in Computer Science or related field plus four (4) years of experience in software engineering or related role
4 years of experience designing, building, and optimizing large-scale backend infrastructure and distributed data systems (e.g., PostgreSQL, MySQL, DynamoDB, Apache Spark, Apache Flink, Apache Kafka) in cloud environments (AWS, GCP, Azure, or equivalent), including cloud-native platforms, core infrastructure components, and optimization techniques (caching, indexing, sharding, replication, transactions, ACID)
4 years of experience with major server-side programming languages and frameworks (e.g., Python, C++, Go, TypeScript)
4 years of experience writing technical design documentation, leading cross-functional projects, and collaborating with cross-functional teams to achieve business impact
3 years of experience developing and maintaining data processing and API systems, including client-server communication frameworks (e.g., gRPC, Thrift)
3 years of experience conducting A/B testing and scientific experimentation (e.g., Statsig, Meta Deltoid, Optimizely) to measure software impact
3 years of experience conducting coding interviews and providing systematic feedback for engineering candidates
2 years of experience with cloud-native tools and infrastructure, such as Docker and Kubernetes
2 years of experience defining and implementing data-driven metrics to support company or team goals
How to Apply: Submit resume and apply online at http://www.fireworks.ai/careers and search for job by title.
Fireworks AI is an equal-opportunity employer. We celebrate diversity and are committed to creating an inclusive environment for all innovators.
Total compensation for this role also includes meaningful equity in a fast-growing startup, along with a competitive salary and comprehensive benefits package. Base salary is determined by a range of factors including individual qualifications, experience, skills, interview performance, market data, and work location. The listed salary range is intended as a guideline and may be adjusted.
Base Pay Range (Plus Equity)
$175,000 — $220,000 USD
Why Fireworks AI?
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.
Similar Jobs
Related searches:
On-site Jobs
Lead Jobs
On-site Lead Jobs
Lead Machine LearningLead AI InfrastructureLead Backend & SystemsLead NLP & Language AILead Generative AILead AI Agents & RAGLead Data Engineering
AI Jobs in New York
Machine Learning in New YorkAI Infrastructure in New YorkBackend & Systems in New YorkNLP & Language AI in New YorkGenerative AI in New YorkAI Agents & RAG in New YorkData Engineering in New York
api-designpytorchagentsmlopsllmdistributed-systemsgenerative-aidata-pipeline
Get jobs like this delivered weekly
Free AI jobs newsletter. No spam.