Amazon.com Inc logo

Senior Inference Engineer, AGI

Amazon.com Inc

  • Sunnyvale, CA
  • 14 days ago

    Highlights

    You will co-design architectures with scientists to make them inference-friendly from inception, own the low-latency streaming serving path, and build the training and reinforcement-learning. real-time runtime that serves it within hard latency budgets, and building the offline systems.

    Numbers & Facts

    LocationSunnyvale, CA
    IndustryRetail
    Company Size10,000 employees or more
    Year Founded1994
    Websitehttp://Amazon.com/militaryroles

    Description

    We are looking for a Senior Inference Engineer to own inference for real-time multimodal

    conversational AI. This is a full-stack inference role: you will work across the entire path a model

    takes from research to production - shaping model architecture so it is servable, building the

    real-time runtime that serves it within hard latency budgets, and building the offline systems

    that train and reinforce it.

    You will operate at the boundary of Science and Inference, taking frontier-scale speech and

    audio models and making them run within real-time latency budgets on production hardware.

    You will co-design architectures with scientists to make them inference-friendly from inception,

    own the low-latency streaming serving path, and build the training and reinforcement-learning

    infrastructure that closes the loop. You will have the compute, data, and runway to solve

    problems that few teams in the world are positioned to tackle.

    As a Senior Engineer, you will own a significant area of the inference stack end to end, drive its

    technical execution, contribute to the team"s roadmap, and work closely with scientists and

    hardware partners to ensure our models run fast enough to feel human in real time - and at a

    cost that makes them viable at scale. You may go deep in one of the areas below while

    contributing across the others.

    Key job responsibilities

    Model Architecture & Inference Co-Design

    • Partner with research scientists to make model architectures servable from inception -

    surfacing the latency, memory, and cost implications of architecture choices before they are

    locked in

    • Implement and optimize the inference path for large-scale multimodal models - attention

    and KV-cache mechanisms, multimodal/autoregressive decoding, and the compute

    primitives on the critical path

    Apply efficiency techniques across the stack - quantization (per-tensor/per-channel/per-

    group, INT8/FP8/BF16), speculative decoding, operator fusion, and paged KV-cache - and

    quantify their quality/latency trade-offs

    • Develop and tune high-performance kernels for critical operations where off-the-shelf

    implementations leave performance on the table, integrating them into production serving

    with minimal overhead

    • Profile end-to-end performance with tools such as Nsight Compute/Systems and roofline

    analysis to identify and eliminate bottlenecks in large-scale inference workloads

    Real-Time & Interactive Runtime

    • Own the real-time serving path for streaming multimodal conversational AI, meeting sub-

    second, streaming latency budgets under concurrent session load

    • Build and tune continuous batching, scheduling, and preemption to balance throughput

    against per-request latency SLAs for interactive workloads

    • Customize production serving frameworks (e.g., vLLM, PyTorch) for real-time streaming

    generative models that fall outside standard LLM serving patterns - sustained low-latency

    output under concurrent session load

    • Implement multi-GPU inference (tensor parallelism, collective communication) for latency-

    critical paths, and drive cost toward parity with existing production baselines

    • Establish latency, throughput, and cost benchmarking, and publish the operational metrics

    that gate deployment

    Offline Systems: Training, RL & Evaluation Infrastructure

    • Build and scale the offline inference systems behind post-training - high-throughput rollout

    generation and reward-model serving for reinforcement learning (RL/RLHF/RLAIF)

    • Ensure train/serve consistency - that the inference path used in RL and evaluation

    faithfully matches production online behavior (e.g., parity across sampling and logit

    processing)

    • Work with the evaluation team to enable offline inference that captures the quality

    dimensions unique to real-time conversation - latency sensitivity, audio quality, and

    interaction naturalness

    About Company

    At Amazon, we don’t wait for the next big idea to present itself. We envision the shape of impossible things and then we boldly make them reality. So far, this mindset has helped us achieve some incredible things. Let’s build new systems, challenge the status quo, and design the world we want to live in. We believe the work you do here will be the best work of your life.

    Wherever you are in your career exploration, Amazon likely has an opportunity for you. Our research scientists and engineers shape the future of natural language understanding with Alexa. Fulfillment center associates around the globe send customer orders from our warehouses to doorsteps. Product managers set feature requirements, strategy, and marketing messages for brand new customer experiences. And as we grow, we’ll add jobs that haven’t been invented yet.

    It’s Always Day 1
    At Amazon, it’s always “Day 1.” Now, what does this mean and why does it matter? It means that our approach remains the same as it was on Amazon’s very first day – to make smart, fast decisions, stay nimble, invent, and stay focused on delighting our customers. In our 2016 shareholder letter, Amazon CEO Jeff Bezos shared his thoughts on how to keep up a Day 1 company mindset. “Staying in Day 1 requires you to experiment patiently, accept failures, plant seeds, protect saplings, and double down when you see customer delight,” he wrote. “A customer-obsessed culture best creates the conditions where all of that can happen.” You can read the full letter here

    Our Leadership Principles
    Our Leadership Principles help us keep a Day 1 mentality. They aren’t just a pretty inspirational wall hanging. Amazonians use them, every day, whether they’re discussing ideas for new projects, deciding on the best solution for a customer’s problem, or interviewing candidates. To read through our Leadership Principles from Customer Obsession to Bias for Action, visit https://www.amazon.jobs/principles

    Similar Jobs

    See more jobs