Lead ML Compiler
5 days ago
About the Role
We are seeking a skilled Compiler Lead to spearhead the development of our next-generation compiler for AI hardware products. The compiler translates code from machine learning frameworks like PyTorch into assembly. You will own the compiler architecture, focus on the backend implementation, and grow the team to ensure production quality.
Key Responsibilities
- Design and develop an ML compiler using MLIR and TVM
- Lead the compiler backend development
- Implement optimization passes
- Optimize workload fragmentation across a multi-core architecture
- Develop codegen techniques for novel instruction set features
- Maintain requirements, solution blueprints, and schedules
- Collaborate with ML and hardware teams on cross-disciplinary projects
What We Offer
- Competitive salary and benefits
- Opportunity to work on groundbreaking technology
- Collaborative and innovative work environment
- Professional growth and development opportunities
- Flexible working hours and remote work options
Requirements
- 5+ years in professional software development
- Experience with ML compilers, specifically MLIR
- Key contributor to previous ML compiler projects
- Proficiency in C++
- Expertise in compiler concepts (e.g., SSA form, optimization passes)
- Familiarity with distributed systems (e.g., MPI)
- Understanding of neural network layers and concepts
- MS or PhD in Computer Science, Electrical Engineering, or related field
- Excellent technical communication and collaboration skills
Desirable Skills
- Knowledge of hardware architecture concepts
- Familiarity with fixed-point quantization
- Experience with other ML compiler frameworks (e.g., TVM)
- Startup experience
- Proficiency in Python and cross-platform development (Windows, Linux)
-
ML Compiler Lead
3 weeks ago
San Bruno, California, United States Acceler8 Talent Full timeAbout UsWe're a cutting-edge technology company that designs chips and AI algorithms for efficient edge device inference. Our innovative approach focuses on distributed, near-memory-compute architecture and sparse mathematics acceleration. Our systems are integrated into various devices, including hearables, smart home appliances, security monitors, and...
-
ML Compiler Architect
2 weeks ago
San Bruno, California, United States Acceler8 Talent Full timeAbout UsWe're a cutting-edge technology company that designs chips and AI algorithms for efficient edge device inference. Our focus is on distributed, near-memory-compute architecture and sparse mathematics acceleration. Our systems are integrated into various devices, including hearables, smart home appliances, and security monitors. Our mission is to...
-
Machine Learning Compiler Engineer
1 week ago
San Jose, California, United States AMD Full timeUnlock the Power of Machine Learning with AMDAt AMD, we're pushing the boundaries of innovation to solve the world's most complex challenges. We're seeking a skilled Machine Learning Compiler Engineer to join our team of talented industry specialists working on developing a cutting-edge machine learning model compiler targeting AMD Inference Accelerator AIE...
-
Machine Learning Compiler Engineer
1 week ago
San Jose, California, United States Advanced Micro Devices , Inc. Full timeTransforming Lives with AMD TechnologyWe're on a mission to build innovative products that accelerate next-generation computing experiences. As a skilled engineer, you'll join a core team of industry specialists working on developing a cutting-edge machine learning model compiler targeting AMD Inference Accelerator AIE hardware devices.The Role:Develop a...
-
Machine Learning Compiler Engineer
5 days ago
San Jose, California, United States AMD Full timeAbout the Role:AMD is seeking a skilled Machine Learning Compiler Engineer to join a core team of talented industry specialists working on developing a cutting-edge machine learning model compiler targeting AMD Inference Accelerator AIE hardware devices.The ideal candidate will be responsible for implementing and improving effective algorithms and techniques...
-
Senior Compiler Engineer
1 month ago
San Jose, California, United States SilicaRise Full timeJob DescriptionWe are seeking a highly skilled Senior Compiler Engineer to join our team at SilicaRise.Responsibilities:Design and implement compilers for DSA, ensuring optimal performance and efficiency.Conduct performance analysis and optimization for ML workloads and kernels, driving innovation in compiler technology.Collaborate with cross-functional...
-
Compiler Performance Engineer
2 weeks ago
San Francisco, California, United States CentML Full timeAbout UsWe're a cutting-edge company, CentML, revolutionizing the AI landscape by making ML models more accessible and affordable. Our mission is to empower anyone to harness the power of AI, and we're committed to making this vision a reality.Our founding team consists of experts in AI, compilers, and ML hardware, with a proven track record of innovation at...
-
San Jose, California, United States Advanced Micro Devices , Inc. Full timeTransforming Lives with AMD TechnologyWe are committed to enriching our industry, communities, and the world through AMD technology. Our mission is to build great products that accelerate next-generation computing experiences - the building blocks for the data center, artificial intelligence, PCs, gaming, and embedded.The Role:We are seeking a talented...
-
San Jose, California, United States AMD Full timeTransforming Lives with AMD TechnologyWe are committed to enriching our industry, communities, and the world through innovative AMD technology. Our mission is to build great products that accelerate next-generation computing experiences – the building blocks for the data center, artificial intelligence, PCs, gaming, and embedded. At AMD, we push the limits...
-
Deep Learning Compiler Engineer for Ryzen AI NPU
2 weeks ago
San Jose, California, United States AMD Full timeAbout the RoleWe are seeking a talented Machine Learning (ML) Compiler Software Engineer to join our growing team in the AI group at AMD. As a key member of our team, you will play a crucial role in developing software tools to deploy cutting-edge ML models on AMD's XDNA Neural Processing Units (NPU). Your primary responsibility will be to design, implement,...
-
San Jose, California, United States Advanced Micro Devices , Inc. Full timeAbout the RoleWe are seeking a talented Machine Learning (ML) Compiler Software Engineer to join our growing team in the AI group and play a crucial role in developing software toolsets to deploy cutting-edge ML models on AMD's XDNA Neural Processing Units (NPU).Key ResponsibilitiesDesign and develop novel algorithms for tiling and mapping quantized ML...
-
Senior Staff Software Engineer
4 weeks ago
San Jose, California, United States AMD Full timeTransforming Lives with AMD TechnologyWe are passionate about using technology to enrich our industry, communities, and the world. Our mission is to build innovative products that accelerate next-generation computing experiences, from the data center to PCs, gaming, and embedded systems. At AMD, we push the limits of innovation to solve the world's most...
-
Software Development Engineer
1 week ago
San Jose, California, United States AMD Full timeTransforming Lives with AMD TechnologyWe're on a mission to build innovative products that accelerate next-generation computing experiences. As a skilled software development engineer, you'll join a talented team working on cutting-edge machine learning model compilers targeting AMD AIE in Ryzen AI PC and embedded FPGA devices.The RoleDevelop a compiler that...
-
Lead AI/ML Data Scientist
4 weeks ago
San Francisco, California, United States Unreal Gigs Full timeJob Title: Lead AI/ML Data ScientistCompany Overview:Welcome to Unreal Gigs, a pioneering company at the forefront of artificial intelligence and machine learning innovation. Our mission is to harness the power of data science to drive transformative change and solve complex problems across industries. We're committed to developing cutting-edge AI and ML...
-
Senior ML Engineering Lead
2 months ago
San Francisco, California, United States Strativ Group Full timeSenior ML Engineering LeadCompensation: up to $450k base + ~1-1.5% equityWe are collaborating with a well-funded startup that is transforming the domain of language models (LLMs) through advanced intelligence and optimization techniques, allowing both businesses and individuals to leverage LLMs tailored to their specific requirements.This organization has...
-
AI/ML Tech Lead
2 weeks ago
San Francisco, California, United States Forge Global Full timeJob Title: AI/ML Tech LeadForge Global is seeking an experienced AI/ML Tech Lead to join our team. As a key member of our engineering team, you will be responsible for leading the development of AI-powered solutions and mentoring junior engineers.Key Responsibilities:Design, develop, and maintain AI/ML platforms that are accurate, secure, and fast.Lead...
-
AI/ML Platform Tech Lead
1 week ago
San Diego, California, United States BioSpace, Inc. Full timeJob SummaryWe are seeking a highly skilled Tech Lead to drive the development and execution of innovative AI/ML-based solutions that advance drug discovery and development at Eli Lilly. As a key member of our team, you will work closely with the Product Owner to co-define delivery milestones for the AI/ML platform and lead the design, development, and...
-
AI/ML Platform Tech Lead
2 weeks ago
San Diego, California, United States BioSpace, Inc. Full timeJob DetailsLilly is a global healthcare leader that unites caring with discovery to make life better for people around the world. Our employees work together to discover and bring life-changing medicines to those who need them, improve the understanding and management of disease, and give back to our communities through philanthropy and volunteerism.What...
-
Tech Lead, AI/ML Specialist
2 weeks ago
San Francisco, California, United States Forge Global Full timeJob Title: Tech Lead, AI/MLAt Forge Global, we're pushing the boundaries of innovation in the private market. As a Tech Lead in AI/ML, you'll play a pivotal role in developing and mentoring our data science team, driving the creation of AI-powered products that transform our business.Key Responsibilities:Design, develop, and maintain a robust AI/ML platform...
-
AI/ML Software Architect
1 week ago
San Jose, California, United States AMD Full timeTransforming Lives with AMD TechnologyWe are seeking a highly skilled AI/ML software architect to join our team at AMD. As a key member of our AI group, you will be responsible for architecting runtime stacks, defining operator mapping and dataflow, and scheduling operators on AMD's XDNA Neural Processing Units.Your work will directly impact the efficiency,...