Talent.com
Software Engineer, AI Infrastructure
Software Engineer, AI InfrastructureFireworks Ai • New York, New York, United States
Software Engineer, AI Infrastructure

Software Engineer, AI Infrastructure

Fireworks Ai • New York, New York, United States
[job_card.30_days_ago]
[job_preview.job_type]
  • [job_card.full_time]
[job_card.job_description]

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 Software Engineer on our AI Infrastructure team, you will help design the core systems that power Fireworks AI’s generative AI platform. You will help build infrastructure and tools that ensure the reliability, performance, quality, and availability of our AI system.

Our mission is to make Fireworks AI the most reliable and user friendly generative AI platform in the world. You will partner closely with our cloud infrastructure team, product team, and performance team to deliver infrastructure that bridges the gap between our customer and the ultra-performant proprietary Fireworks inference engine.

Key Responsibilities :

  • Contribute to the design and development of scalable backend infrastructure that supports distributed training, inference, and data pipelines
  • Build and maintain core backend services such as LLM CI / CD pipeline, control plane, and model serving systems
  • Support performance optimization, cost efficiency, and reliability improvements across compute, storage, and networking layers
  • Building frameworks and safeguards to ensure Fireworks AI has the best model quality in the industry
  • Collaborate with performance, training, and product teams to translate research and product needs into infrastructure solutions
  • Participate in code reviews, technical discussions, and continuous integration and deployment processes

Minimum Qualifications :

  • Bachelor’s degree in Computer Science, Engineering, or a related technical field (or equivalent practical experience)
  • 3 years of experience in software engineering, with a focus on infrastructure or machine learning systems
  • Strong programming skills in Python, Go, or a similar language
  • Proven experience in ML infrastructure and tooling (e.g., PyTorch, MLflow, Vertex AI, SageMaker, Kubernetes, etc.).
  • Basic understanding of LLM knowledge (e.g., context length, disaggregated prefill, KV cache memory estimation, etc)
  • Preferred Qualifications :

  • 5+ years of experience in software engineering, with a focus on infrastructure or machine learning systems
  • Experience with open source inference engine like vLLM, Sglang, or TRT-LLM
  • Contributions to open-source infrastructure or ML projects
  • Experience in building large scale ML / MLOps infrastructure
  • 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.

    [job_alerts.create_a_job]

    Software Engineer Infrastructure • New York, New York, United States

    [internal_linking.related_jobs]
    Software and AI Engineer

    Software and AI Engineer

    CoreWeave • New York City, New York, USA
    [job_card.full_time] +1
    CoreWeave is The Essential Cloud for AI.Built for pioneers by pioneers CoreWeave delivers a platform of technology tools and teams that enables innovators to build and scale AI with confidence.Trus...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    AI Infrastructure Engineer, Core Infrastructure

    AI Infrastructure Engineer, Core Infrastructure

    Scale AI, Inc. • New York, NY, United States
    [job_card.full_time]
    As a Software Engineer on the ML Infrastructure team, you will design and build the next generation of foundational systems that power all ML Infrastructure compute at Scale - from model training a...[show_more]
    [last_updated.last_updated_30] • [promoted]
    AI Platform Engineer Lead

    AI Platform Engineer Lead

    Capital Group • New York, NY, United States
    [job_card.full_time]
    I can succeed as a Lead AI Platform Engineer at Capital Group.As a Lead AI Platform Engineer, you will serve as a technical SME and lead workstreams to design, build, and operate the foundational c...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    AI Engineer

    AI Engineer

    OneStream Software • New York, NY, United States
    [job_card.full_time]
    Range applies to US candidates only) + Benefits / Variable Comp / Equity - Range may vary based on experience.Vision, Medical, Life, Dental, 401K. This role requires the candidate to work from one of th...[show_more]
    [last_updated.last_updated_30] • [promoted]
    AI Infrastructure Engineer, Core Infrastructure

    AI Infrastructure Engineer, Core Infrastructure

    Scale AI • New York, NY, United States
    [job_card.full_time]
    AI Infrastructure Engineer, Core Infrastructure.As a Software Engineer on the ML Infrastructure team, you will design and build the next generation of foundational systems that power all ML Infrast...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    Software Engineer III-AI Platform Engineer

    Software Engineer III-AI Platform Engineer

    JPMorganChase • Jersey, New Jersey, USA
    [job_card.full_time]
    We have an exciting and rewarding opportunity for you to take your software engineering career to the next level.As a Software Engineer III at JPMorgan Chase within the Corporate Sector Infrastruct...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    Senior AI Software Engineer NYC AI Data Science Labs – Investment Management

    Senior AI Software Engineer NYC AI Data Science Labs – Investment Management

    J Harlan Group • New York City, New York, USA
    [job_card.full_time]
    NYC AI Data Science Labs Investment Management.J Harlan Group is currently conducting a search for a Senior AI Software Engineer at a prominent Investment Manager in their NYC AI Data Science Labs...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    Senior AI / ML Software Engineer (Remote in North Carolina)

    Senior AI / ML Software Engineer (Remote in North Carolina)

    Rocket Lawyer • New York, New York, United States
    [filters.remote]
    [job_card.full_time]
    We believe everyone deserves access to affordable and simple legal services.Founded in 2008, Rocket Lawyer is the largest and most widely used online legal service platform in the world.With office...[show_more]
    [last_updated.last_updated_30] • [promoted]
    Senior AI / ML Software Engineer – Remote

    Senior AI / ML Software Engineer – Remote

    UnitedHealth Group • New York, New York, US
    [filters.remote]
    [job_card.full_time]
    Optum is a global organization that delivers care, aided by technology to help millions of people live healthier lives.The work you do with our team will directly improve health outcomes by connect...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    Senior AI Engineer - Cloud Engineering Focus (Hybrid)

    Senior AI Engineer - Cloud Engineering Focus (Hybrid)

    OneStream Software • New York, NY, United States
    [job_card.full_time]
    Senior AI Engineer - Cloud Engineering Focus.Hybrid, USA (Birmingham, MI, Chicago, IL, New York, NY, Fort Lauderdale, FL). Range applies to US candidates only) + Benefits / Variable Comp / Equity - Rang...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    AI Infrastructure Engineer, Model Serving Platform

    AI Infrastructure Engineer, Model Serving Platform

    Scale AI • New York, NY, United States
    [job_card.full_time]
    As a software engineer on the ML Infrastructure team, you will work on developing the platform for orchestrating post-training and model evaluation jobs. At Scale, we are constantly developing new d...[show_more]
    [last_updated.last_updated_30] • [promoted]
    AI Engineer

    AI Engineer

    Metropolitan Commercial Bank • New York, New York, United States
    [job_card.full_time]
    Metropolitan Commercial Bank (the “Bank”) is a full-service commercial bank based in New York City.The Bank provides a broad range of business, commercial, and personal banking products and service...[show_more]
    [last_updated.last_updated_30] • [promoted]
    AI Software Engineer

    AI Software Engineer

    VirtualVocations • Yonkers, New York, United States
    [job_card.full_time]
    A company is looking for an AI Software Engineer I to help build core features and pipelines for their AI platform.Key Responsibilities Build AI-enabled product features such as chat, recommendat...[show_more]
    [last_updated.last_updated_30] • [promoted]
    Distinguished AI Engineer (Agentic AI Platform Infrastructure)

    Distinguished AI Engineer (Agentic AI Platform Infrastructure)

    Capital One • New York, New York, United States
    [job_card.full_time] +1
    Distinguished AI Engineer (Agentic AI Platform Infrastructure) At Capital One, we are creating responsible and reliable AI systems, changing banking for good. For years, Capital One has been an indu...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    Cloud & AI Applications Engineer

    Cloud & AI Applications Engineer

    Map Ssg • New York City, New York, USA
    [job_card.full_time]
    The Enterprise Applications team is seeking a motivated technically savvy and business-oriented professional to serve as the initial point of technical contact for the firms cloud-based and AI-powe...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    Lead Engineer, Agentic AI

    Lead Engineer, Agentic AI

    VirtualVocations • Yonkers, New York, United States
    [job_card.full_time]
    A company is looking for a Lead Engineer in Agentic AI.Key Responsibilities Own end-to-end technical outcomes including architecture, design, implementation, and production support Design agenti...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    AI Software Engineer NYC AI Data Science Labs – Investment Management

    AI Software Engineer NYC AI Data Science Labs – Investment Management

    J Harlan Group • New York City, New York, USA
    [job_card.full_time]
    NYC AI Data Science Labs Investment Management.J Harlan Group is currently conducting a search for a AI Software Engineer at a prominent Investment Manager in their NYC AI Data Science Labs.This o...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]
    Cloud & AI Applications Engineer

    Cloud & AI Applications Engineer

    Cleary Gottlieb Steen & Hamilton LLP • New York City, New York, USA
    [job_card.full_time]
    Cleary Gottlieb is a pioneer in globalizing the legal profession.We have 14 offices in major financial centers around the world but we operate as a single integrated global partnership and not as a...[show_more]
    [last_updated.last_updated_variable_days] • [promoted]