System Software Engineer - AI

AI EngineerMachine Learning EngineerFull TimeRemote

Location

United States

Posted

4 days ago

Salary

$140K - $200K / year

C++PythonLinuxCUDAPy TorchJAXDeep SpeedV LLMSglangTRT LLMKubernetesTelemetry

Job Description

System Software Engineer - AI

About us:

We are a stealth-mode startup building foundational technology to address performance, scalability, and resiliency challenges in large-scale AI data center clusters. We are backed by top-tier VC firms and notable angel investors.

The company is led by experienced builders and operators who have founded companies, taken them to scale, and exited successfully. We work with a strong sense of unity and shared responsibility, and we expect trust, integrity, and respect in how we collaborate and make decisions. We hold ourselves accountable to one another and to the quality of the work we deliver.

Headquartered in Silicon Valley, we operate across a mix of remote and on-site locations in the U.S. and Canada. We aim to create an environment where people are treated fairly, supported in their growth, and are empowered to do meaningful work alongside others who take the craft seriously.

We are looking for:

We are looking for a talented System Software Engineer to help us redefine the infrastructure layer of AI. In this role, you will bridge the gap between high-level AI frameworks and low-level system software. You will be responsible for designing and implementing the communication and execution primitives that allow large-scale AI models to run efficiently across thousands of GPUs. We are looking for a "builder" who thrives in the early stages of a product’s lifecycle and is passionate about solving the "hard" systems problems of the generative AI era.

Key Responsibilities:

  • Collaborate across the stack to influence the design of our foundational technology, ensuring it meets the needs of next-generation AI models.

  • Identify and resolve performance bottlenecks in distributed training and inference workloads through deep-dive analysis of the software-hardware interface.

  • Conduct rigorous performance benchmarking and characterization on multi-node clusters.

Required Skills and Qualifications:

  • Strong proficiency in C++ and Python, with a deep understanding of systems programming fundamentals (memory management, concurrency, OS internals).

  • Proficient in a Linux development environment.

Desired Skills:

  • Experience with GPU programming (CUDA) and performance optimization for parallel architectures.

  • Familiarity with distributed AI frameworks (PyTorch, JAX, or DeepSpeed) and/or inference engines (vLLM, SGLang, Dynamo/TRT-LLM).

  • Hands-on experience with large-scale cluster orchestration and telemetry tools.

Education:

  • Bachelor's or Master's degree in Computer Engineering, Computer Science, or a related field.

Compensation:

Target base salary for this role is $140,000 - $200,000 per year + meaningful equity + benefits + 401k. Our salary ranges are determined by role, level, experience, and location.

#LI-EW1

Related Job Pages

More AI Engineer Jobs

AI Engineer4 days ago
Full TimeRemoteTeam 5,001-10,000H1B Sponsor

The Lead Engineer will own the technical architecture, design, and implementation for new AI-driven insurance products, translating open-ended opportunities into defined system designs and production-grade solutions. This involves rapid build-test-learn cycles, developing features using modern AI frameworks, and establishing best practices for engineering excellence.

PythonLLMAWSSystem DesignCI/CDObservabilityVector DatabasesEmbeddingsPrompt EngineeringWorkflow Automation
United States
$106K - $180K / year

Applied AI Engineer

Pure Storage

Helping innovators uncomplicate data storage, forever.

AI Engineer4 days ago
Full TimeRemoteTeam 1,001-5,000Since 2010H1B Sponsor

Applied AI Engineer leading customer engagements for enterprise AI at Pure

CloudKafkaKubernetesPyTorchRaySpark
United States
$171.5K - $257.6K / year

Director, AI Engineering

brightwheel

#1 platform for early education

AI Engineer4 days ago
Full TimeRemoteTeam 201-500Since 2014H1B Sponsor

This role involves leading engineering teams to deliver AI-powered product and platform capabilities aimed at reducing administrative toil, improving responsiveness, and surfacing better insights for early education operators. The director will build systems and AI-native ways of working to transform the platform from a system of record into a system of action, accelerating value delivery to customers.

United States
$215K - $303K / year

AI Engineer (Intern)

Juniper Square

Where partnerships drive potential.

AI Engineer4 days ago
Full TimeRemoteTeam 201-500H1B No Sponsor

The intern will collaborate on designing and prototyping AI-powered features, contributing to the development, training, and evaluation of machine learning and NLP models using industry-standard tools. They will also work with large datasets, apply foundational ML techniques, and assist in integrating AI models into production systems.

PythonTensorFlowPyTorchscikit-learnNumPyPandasMachine LearningDeep LearningNLPLLMRESTGraphQLSQLNoSQLAWSGCPGit
United States