Full-Time

AI GPU Performance Engineer

Confirmed live in the last 24 hours

Modular

Modular

51-200 employees

AI developer platform for unified hardware and frameworks

AI & Machine Learning

Compensation Overview

$234,000 - $319,000Annually

+ Bonus + Equity

Senior

Remote in USA + 1 more

Required Skills
CUDA
Requirements
  • 3+ years of relevant experience working on complex code and systems.
  • Experience with GPU programming languages such as CUDA or OpenCL.
  • Experience with GPU assembly such as PTX and SASS.
  • Strong understanding of GPU architectures and performance optimizations.
  • Experience with AI workloads and performance tuning considerations such as fusion strategies;
  • Experience with core AI kernels such as matrix multiply and convolution.
  • Strong familiarity with using GPU profilers, debuggers, etc.
  • Deep interest in machine learning technologies and use cases.
  • Creativity and curiosity for solving complex problems, a team-oriented attitude that enables you to work well with others, and alignment with our culture.
Responsibilities
  • Design, develop, and optimize high-performance AI numeric and data manipulation kernels/operators for GPUs.
  • Achieve state-of-the-art performance by leveraging software and micro-architectural features of GPUs.
  • Work with compiler, framework, and runtime teams to deliver end-to-end performance that fully utilizes GPU workstations and servers.
  • Collaborate with machine learning researchers to guide system development for future ML trends.

The company offers a next-generation AI developer platform that unifies AI frameworks and hardware, providing unparalleled performance and cost savings. The platform includes a new programming language, Mojo, and a modular engine that enables model deployment, compute portability, and scalability, revolutionizing AI infrastructure.

Company Stage

Series A

Total Funding

$130M

Headquarters

null, null

Founded

2022

Growth & Insights
Headcount

6 month growth

6%

1 year growth

19%

2 year growth

547%