Majestic Labs AI Logo

Majestic Labs AI

Compiler Engineer (PyTorch / Triton / LLVM / MLIR)

Posted 29 Days Ago
Be an Early Applicant
In-Office
Los Altos, CA, USA
Senior level
In-Office
Los Altos, CA, USA
Senior level
Design, develop, and optimize compiler components for AI workloads across CPUs, GPUs, and custom accelerators. Enhance PyTorch Inductor, Triton, LLVM, and MLIR toolchains; create custom intermediate representations, code-generation passes, and optimization strategies; tune computational kernels for performance; and contribute to open-source compiler projects.
The summary above was generated by AI
Description

We’re a fast-moving, US-Israeli AI startup building next-generation infrastructure for the world’s most demanding AI workloads. Our mission is to accelerate the future of intelligence and make it ubiquitously accessible by delivering full stack custom AI servers optimized for large-scale inference and training.

Backed by top-tier investors and led by industry veterans, we’re scaling rapidly. We seek forward-thinking engineers and operators who thrive in a collaborative environment and excel at solving complex challenges from first principles. If you want to build infrastructure that fundamentally changes what the world can accomplish with AI, come join us!

About the position



We are looking for an experienced, highly skilled and motivated Senior Compiler Engineer to join our compiler team. In this role, you will be responsible for designing, developing, and maintaining critical components of our AI-driven compilation stack. You will work on PyTorch, Triton, LLVM, and MLIR to build robust, scalable, and high-performance solutions for diverse

hardware backends. This is a hands-on technical position where you will solve complex problems, optimize for performance, and contribute to the next generation of our technology.

Responsibilities



  • Design, Implement, and Optimize Compiler Components: Architect and develop critical compiler modules to efficiently translate and optimize AI models for deployment across a variety of hardware platforms, including CPUs, GPUs, and emerging custom accelerators
  • Enhance and unify PyTorch Inductor, Triton, LLVM, and MLIR toolchains to support cutting-edge architectures, facilitate seamless interoperability, and enable rapid experimentation with new compiler features
  • Create and maintain custom IRs, code generation passes, and optimization strategies tailored for AI workloads, focusing on both general and domain-specific improvements
  • Profile and tune computational kernels—such as linear algebra operations, matrix multiplications, and elementwise computations—to achieve optimal performance, scalability, and resource efficiency on diverse hardware
  • Open-Source Engagement: Actively contribute to the LLVM, MLIR, Triton, and PyTorch open-source projects, sharing improvements, collaborating with the developer community, and driving the evolution of the AI compiler ecosystem.
Requirements
  • Bachelor’s or Master’s in Computer Science, Computer Engineering, or a related field from a recognized university
  • 7+ years experience in compiler engineering or closely related fields
  • Deep knowledge of LLVM and MLIR internals, including IR transformations, code generation, and backend optimization techniques
  • Experience with PyTorch (Inductor/Dynamo) and Triton, including their compiler subsystems
  • Proven experience and expertise in C/C++ programming
  • Demonstrated expertise in performance optimization, including vectorization, parallelization, and hardware-specific tuning
  • Advanced debugging, analytical, and system-level thinking skills
  • Excellent communication skills with a strong track record of cross-functional collaboration

Preferred Qualifications



  • Strong understanding of AI models - both training and inference pipelines
  • Experience developing compiler support for custom hardware accelerators, including ASICs, FPGAs, or novel AI chips
  • Active contributions to open-source compiler frameworks, demonstrating leadership and community involvement
  • Familiarity with distributed training strategies, graph compilers, and advanced memory models


We seek smart, curious, self-driven individuals who enjoy creative problem-solving and continuous learning. Even if you don't meet all requirements, we'd love to hear from you if you have the mindset to tackle complex challenges.

Majestic Labs is an equal opportunity employer deeply committed to diversity of background and thought. If you are ready to dare greatly, learn from mistakes, engage in vigorous debates, and help create a step-function shift in human technical capability, we want to meet you!

Similar Jobs

4 Minutes Ago
In-Office
Palo Alto, CA, USA
158K-237K Annually
Mid level
158K-237K Annually
Mid level
Artificial Intelligence • Big Data • Cloud • Information Technology • Software • Cybersecurity • Data Privacy
Designs and develops backend solutions for Disaster Recovery as a Service across on-premises and cloud platforms. Builds scalable application recovery features for AWS, Azure, and GCP, writes unit tests, solves complex distributed-systems problems, and collaborates with product managers, designers, and engineers to deliver reliable products.
Top Skills: AWSAzureC++Google Cloud PlatformJavaScalaVMware
12 Minutes Ago
Easy Apply
In-Office
Easy Apply
80K-100K Annually
Mid level
80K-100K Annually
Mid level
Digital Media • eCommerce • Information Technology • Marketing Tech • Pet • Retail • Social Media
Develop and direct high-performing paid social video ads for dog and cat food brands. Responsibilities include generating concepts, writing direct-response scripts, directing talent and shoots, collaborating with editors, analyzing Meta performance data, iterating creative, and maintaining a continuous testing pipeline. The role uses AI tools for ideation and optimization and requires hybrid office and on-set availability in Burbank, plus periodic travel shoots.
Top Skills: Ai ToolsMeta
21 Minutes Ago
Remote or Hybrid
San Francisco, CA, USA
154K-257K Annually
Senior level
154K-257K Annually
Senior level
Artificial Intelligence • Information Technology • Machine Learning • Natural Language Processing • Productivity • Software • Generative AI
Build Superhuman Mail’s AI platform, including MCP tools and standards, agentic workflows, personalization, LLM evaluation systems, and model-selection strategies. The role involves shipping production LLM features, partnering across teams on reusable AI capabilities, and potentially shaping data collection and modeling approaches. Engineers will work primarily with backend systems and applied AI while adapting quickly to evolving technologies and product priorities.
Top Skills: AnthropicAWSGoKubernetesLlmsMcpOpenaiPostgresPythonRedisTurbopufferTypescript

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account