Founding Senior AI Infrastructure Engineer

Posted Yesterday
Be an Early Applicant
Palo Alto, CA, USA
In-Office
Entry level
Artificial Intelligence • Machine Learning • Software • Generative AI
The Role
Build and optimize foundational AI infrastructure for large-scale training, reinforcement learning, rollout, and inference workloads. Responsibilities include profiling bottlenecks, optimizing GPU and distributed execution, developing scheduling and load-balancing systems, modeling performance and capacity, improving fault tolerance, and creating benchmarks and observability tools. The role partners closely with AI researchers and systems engineers to improve throughput, reliability, utilization, latency, and cost across complex GPU clusters.
Summary Generated by Built In
About us

We’re building toward a world where every company can become its own AI lab.
Goaly is a stealth AI startup founded by ex-Meta MSL engineers and researchers. Our mission is to dramatically lower the cost, time, and talent barriers to building proprietary AI — and make each generation of models faster and cheaper to build than the last.

Backed by leading AI investors and endorsed by frontier AI researchers and builders, we’re looking for exceptional new grads who want to work on hard, foundational AI systems problems with outsized ownership from day one.

About the role

Running modern AI workloads at scale creates systems problems that rarely fit within a single layer of the stack. A slowdown that appears in a training job may originate in a GPU kernel, collective communication, data movement, container runtime, storage path, scheduler, or interaction between model architecture and hardware topology.

You will identify these bottlenecks and build systems that improve the throughput, efficiency, and robustness of our largest distributed workloads. Your scope will span post-training, agentic reinforcement learning, model training, rollout inference, and the GPU cluster platform beneath them. You will work closely with researchers and systems engineers, develop a quantitative understanding of performance, and turn one-off investigations into durable infrastructure improvements.

This role is a strong fit for an exceptional systems or performance engineer who wants to work on frontier AI infrastructure. Deep ML experience is valuable but not required; we care most about a track record of solving difficult systems problems at scale and the ability to become fluent in new parts of the ML stack quickly.

What you'll do
  • Profile end-to-end AI workloads and identify limiting resources across model code, GPU kernels, memory, collective communication, networking, storage, orchestration, and environment execution.

  • Build low-latency, high-throughput sampling and inference systems for large language models, including batching, scheduling, caching, load balancing, and efficient weight updates.

  • Optimize GPU execution through kernel and graph profiling, memory-layout improvements, reduced-precision computation, communication overlap, compilation, and targeted CUDA or Triton work.

  • Improve distributed training and reinforcement-learning performance across heterogeneous GPU and CPU workloads, variable-length rollouts, complex network topologies, and changing model architectures.

  • Design quantitative performance and capacity models that predict bottlenecks, explain scaling behavior, guide hardware and topology choices, and prioritize engineering work.

  • Build scheduling and load-balancing mechanisms that improve accelerator utilization while respecting memory, locality, topology, latency, and fault-domain constraints.

  • Design fault-tolerant distributed systems that detect failures early, isolate their impact, recover efficiently, and preserve correctness during long-running jobs.

  • Investigate difficult production issues such as kernel-level stalls, network-latency spikes, collective timeouts, memory fragmentation, stragglers, and performance regressions in containerized environments.

  • Develop benchmarks, profiling tools, performance dashboards, and regression tests that make system behavior visible and allow improvements to be measured under realistic workloads.

  • Partner with researchers to understand new models and algorithms, remove infrastructure constraints from the experimental loop, and translate successful optimizations into reusable platform capabilities.

What success looks like
  • Within your first three months, you have built a quantitative understanding of at least one critical workload, identified its dominant bottlenecks, and shipped a measurable performance or reliability improvement.

  • Within six to twelve months, you have delivered sustained gains in throughput, accelerator utilization, latency, scaling efficiency, or cost across real training, rollout, or inference workloads.

  • Performance investigations become faster and more rigorous because the team has better benchmarks, models, profiles, and observability—not just undocumented fixes.

  • Large distributed jobs run predictably across complex hardware and network topologies, and failures or regressions can be detected, explained, and recovered from with minimal researcher intervention.

You may be a good fit if you have
  • Significant software-engineering, distributed-systems, high-performance-computing, or ML-infrastructure experience, particularly with performance-critical systems operating at large scale.

  • Exceptional programming and debugging ability in Python and at least one systems language such as C++, Rust, or Go.

  • Strong systems fundamentals, including operating systems, concurrency, memory, networking, storage, scheduling, containerization, and failure handling.

  • A track record of using measurement to solve ambiguous performance problems: forming hypotheses, designing representative benchmarks, reading profiles and traces, identifying root causes, and validating improvements under production conditions.

  • The ability to reason across abstraction boundaries, from model architecture and framework execution to accelerator behavior, distributed runtimes, cluster topology, and infrastructure services.

  • A results-oriented mindset, flexibility about where in the stack to work, and a willingness to take ownership beyond a narrowly defined job description.

  • Strong communication and collaboration skills, including the ability to work directly with researchers, explain complex systems behavior clearly, and turn repeated investigations into maintainable tools and abstractions.

  • Interest in developing deep expertise in machine learning systems, even if your prior work has been primarily in distributed systems, HPC, operating systems, compilers, databases, or networking.

Strong pluses
  • Experience building or operating high-performance, large-scale ML training, post-training, or inference systems.

  • Experience with GPU or accelerator programming, including CUDA, Triton, custom kernels, low-precision computation, memory optimization, or compiler stacks.

  • Familiarity with ML framework internals and distributed stacks such as PyTorch, JAX, torch.distributed, FSDP, Megatron, DeepSpeed, Ray, vLLM, SGLang, or TensorRT-LLM.

  • Knowledge of NCCL or RCCL, RDMA, InfiniBand, RoCE, NVLink, collective algorithms, network topology, or topology-aware workload placement.

  • Experience with Linux or operating-system internals, container runtimes, Kubernetes, cluster schedulers, storage systems, or production observability.

  • Familiarity with transformer architectures, language-model training, reinforcement learning, online sampling, or the performance characteristics of mixture-of-experts models.

  • Meaningful contributions to open-source systems, ML frameworks, compilers, kernels, networking software, or performance tooling.

How we work
  • Mission first. We choose work for its impact on the mission and take responsibility for the outcome, not just our assigned tasks.

  • High agency. We identify what is missing, form a plan, and move without waiting for perfect clarity.

  • Speed with rigor. We ship, measure, and iterate quickly while protecting correctness, safety, and reliability.

  • Flexible scope. We cross team and technical boundaries when that is the fastest way to solve the real problem.

  • Low ego, high standards. We give direct feedback, change our minds when the evidence changes, and help the whole team win.

  • Continuous learning. The stack changes quickly; we are willing to learn unfamiliar systems, methods, and domains as the work demands.

A note on qualifications. We care more about exceptional evidence than a perfect keyword match. If the work excites you and you can show unusual strength, learning speed, or ownership, we encourage you to apply even if your background does not match every preferred qualification.

Equal opportunity

We are an equal opportunity employer. We consider qualified applicants without regard to race, color, religion, sex, sexual orientation, gender identity or expression, national origin, age, disability, veteran status, genetic information, or any other characteristic protected by applicable law. We provide reasonable accommodations for candidates who need them during the hiring process.

Skills Required

  • Significant software engineering, distributed systems, high-performance computing, or ML infrastructure experience
  • Exceptional programming and debugging ability in Python and at least one systems language such as C++, Rust, or Go
  • Strong fundamentals in operating systems, concurrency, memory, networking, storage, scheduling, containerization, and failure handling
  • Demonstrated ability to solve ambiguous performance problems using benchmarks, profiles, traces, hypotheses, and production validation
  • Ability to reason across model architecture, framework execution, accelerators, distributed runtimes, cluster topology, and infrastructure services
  • Strong communication and collaboration skills with researchers and systems engineers
  • Experience with high-performance large-scale ML training, post-training, or inference systems
  • GPU or accelerator programming experience, including CUDA, Triton, custom kernels, low-precision computation, memory optimization, or compiler stacks
  • Familiarity with ML frameworks and distributed stacks such as PyTorch, JAX, FSDP, Megatron, DeepSpeed, Ray, vLLM, SGLang, or TensorRT-LLM
  • Knowledge of NCCL or RCCL, RDMA, InfiniBand, RoCE, NVLink, collective algorithms, or topology-aware workload placement
  • Experience with Linux internals, container runtimes, Kubernetes, cluster schedulers, storage systems, or production observability
  • Familiarity with transformer architectures, language-model training, reinforcement learning, online sampling, or mixture-of-experts performance
  • Meaningful contributions to open-source systems, ML frameworks, compilers, kernels, networking software, or performance tooling
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
80 Employees
Year Founded: 2025

What We Do

Goaly is building AI systems and a platform that make custom, domain-specific AI more affordable for every business. Its work spans model post-training, reinforcement-learning infrastructure, large-scale training, reliable inference, and product engineering. The company aims to help enterprises achieve strong task performance at a fraction of typical costs, while developing agents that can reason, use tools, and complete meaningful real-world work.

Similar Jobs

CDW Logo CDW

Technical Writer

Information Technology
Remote or Hybrid
US
15100 Employees
64K-89K Annually

CDW Logo CDW

Customer Success Manager

Information Technology
Hybrid
Vernon, CA, USA
15100 Employees
95K-100K Annually

Liberty Mutual Insurance Logo Liberty Mutual Insurance

Global Private Equity Director

Artificial Intelligence • Fintech • Insurance • Marketing Tech • Software • Analytics
Hybrid
8 Locations
40000 Employees
137K-257K Annually

Liberty Mutual Insurance Logo Liberty Mutual Insurance

Senior Field Property Claim Resolution Specialist, Large Loss

Artificial Intelligence • Fintech • Insurance • Marketing Tech • Software • Analytics
Hybrid
8 Locations
40000 Employees
86K-157K Annually

Similar Companies Hiring

Revel Thumbnail
Aerospace • Hardware • Robotics • Software
Marina Del Rey, California
60 Employees
Blee Thumbnail
Artificial Intelligence • Marketing Tech • Software
New York, New York
30 Employees
Vega Thumbnail
Artificial Intelligence • Automotive • Insurance • Transportation
US
43 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account