Systems/GPU Research Engineer

Vast.ai Systems/GPU EngineerVast.ai's cloud powers AI projects and businesses all over the world. We are democratizing and decentralizing AI computing—reshaping our future for the benefit of humanity. We are a growing and highly motivated team dedicated to an ambitious technical plan. Our structure is flat, our ambitions are out-sized, and leadership is earned by shipping excellence. We seek engineers with strong intrinsic drive, a true passion for advancing the state of the art, and a mix of architecture, coding, and communication skills.LOCATION: On-site at our office in San Francisco or Westwood, Los Angeles.About the RoleAs 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-TimeOn-site at either our SF or LA officesTech StackC++/CUDA, GPGPU, Python, LinuxIdeal ExperienceExpertise in systems engineering across the tech stackDeep understanding of GPU architecturesStrong holistic background in neural network performance and toolingPublished research at top AI conferencesKey ResponsibilitiesDevelop or extend parallel generic GPU libraries and kernelsHelp design and deploy market-based resource management systemsQuickly investigate and summarize options for new system architecturesPrototype and evaluate novel state-of-the-art methods/modelsInvestigate and learn new frameworks and toolsInterview ProcessAfter 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.BenefitsComprehensive health, dental, vision, and life insurance401(k) with company matchMeaningful early-stage equityOnsite meals, snacks, and close collaboration with founders/tech leadersAmbitious, fast-paced startup culture where initiative is rewarded