Systems/GPU Research Engineer

Vast.ai Inc

$120K — $180K *
Technical Services
Less than 5 years of experience
Job Overview by Ladders

Qualifications

  • 5+ years in systems engineering
  • Strong expertise in GPU architectures and programming
  • Proficient in C++/CUDA, Python, and Linux
  • Demonstrated experience with neural network performance
  • Published author in top-tier AI conferences

Responsibilities

  • Develop and optimize parallel GPU libraries and kernels
  • Design and implement resource management systems
  • Research and summarize new system architecture options
  • Prototype and assess innovative AI methods
  • Learn and evaluate new frameworks and tools

Benefits

  • Comprehensive health, dental, vision, and life insurance
  • 401(k) with company match
  • Equity in early-stage startup
  • Free onsite meals and snacks
  • Collaborative environment with founders and tech leaders
  • Dynamic, initiative-driven startup culture
Full Job Description

LOCATION: On-site at our office in San Francisco or Westwood, Los Angeles.
About the Role

As a systems/GPU engineer, you will play a crucial role in developing new kernels and algorithms that can improve inference for AI models. You will help develop new high-performance tensor libraries and auto-optimization tools. Collaborating directly with our technical founder and diverse team, you will enhance the performance and efficiency of our AI systems. Your ability to research and stay on top of cutting-edge papers will be vital in staying up-to-date with the latest advancements in AI model inference and GPU programming techniques.
  • Full-Time
  • On-site at either our SF or LA offices
Tech Stack

C++/CUDA, GPGPU, Python, Linux
Ideal Experience
  • Expertise in systems engineering across the tech stack
  • Deep understanding of GPU architectures
  • Strong holistic background in neural network performance and tooling
  • Published research at top AI conferences
Key Responsibilities
  • Develop or extend parallel generic GPU libraries and kernels
  • Help design and deploy market-based resource management systems
  • Quickly investigate and summarize options for new system architectures
  • Prototype and evaluate novel state-of-the-art methods/models
  • Investigate and learn new frameworks and tools
Interview Process

After submitting your application, our technical team reviews your credentials. If selected, you'll proceed through the following stages:
  • Initial screening (virtual, 15 minutes)
  • Quick dive into Vast, systems and architectures (virtual, 30 minutes)
  • LLM-assisted coding assessment (virtual, 1 hour)
  • Meet and greet with coding assessment (on-site, 2 hours)
Our goal is to complete the interview process in two weeks.
Benefits
  • Comprehensive health, dental, vision, and life insurance
  • 401(k) with company match
  • Meaningful early-stage equity
  • Onsite meals, snacks, and close collaboration with founders/tech leaders
  • Ambitious, fast-paced startup culture where initiative is rewarded

Similar Jobs

More Jobs at Vast.ai Inc

More Technical Services Jobs

Find similar Systems/GPU Research Engineer jobs: