AI Compiler Engineer – New College Grad

This is a fully remote position, open to applicants in California, +2 more states.

📋 Description

• Create innovative technologies in the realm of machine learning compilers and AI systems.

• Design cutting-edge AI compiler technologies tailored for NVIDIA hardware architecture.

• Establish new abstractions for AI compilers.

• Develop solutions that enhance the efficiency of machine learning system engineering.

• Optimize significant machine learning workloads for better performance.

• Drive innovation by developing advanced technologies in machine learning compilers and systems.

• Design, implement, and refine compilers specifically for impactful AI workloads.

• Construct agent-driven solutions to streamline the entire development process.

• Collaborate in the co-design of agent system solutions alongside contemporary and future AI compiler technologies.

• Work closely with various NVIDIA engineering teams to create high-impact solutions aimed at accelerating machine learning.

⛳️ Requirements

• A Bachelor's degree in Computer Science, Electrical Engineering, or a related discipline (or comparable experience).

• Proven experience in the development of machine learning systems, including ML compilers, LLM inference kernels, and kernel generation.

• Proficiency in Python, C/C++, and GPU programming.

• Expertise in machine learning compilers like Apache TVM and MLIR.

• A background in creating AI agent-driven solutions and automation processes.

• Specialized knowledge in domain-specific compiler and library solutions for LLM inference and training, such as FlashInfer and Flash Attention.

• Extensive experience in machine learning systems research and product development.

• Ownership of or contributions to open source projects.

🏝️ Benefits

• Equity.

• Comprehensive benefits package.