Design and build core components of the ML compiler, focusing on optimization strategies, compiler architecture, and collaboration with hardware teams.
You will design and build core components of our ML compiler, owning critical parts of the transformation and optimization pipeline.
Requirements
Benefits
Your work will directly determine whether models fit within hardware constraints and achieve real-time performance.
- Design and implement compiler architecture across:
- IR design and transformation pipelines
- Optimization strategies for scheduling, tiling, and memory reuse
- Lead development of complex compiler passes such as:
- Global memory allocation and liveness-driven reuse
- Cross-operator fusion and graph partitioning
- Hardware-aware scheduling strategies
- Develop cost models and optimization heuristics
- Explore advanced techniques:
- Constraint-based optimization (e.g., ILP/MILP/CP)
- Scheduling optimization
- Drive debugging of system-level issues (correctness, performance, HW mismatches)
- Collaborate with hardware teams on co-design of abstractions and execution models
Requirements
Required qualifications and experience:
- 4+ years of experience in compilers, systems, or performance engineering
- Masters or PhD in Computer Science, Electrical Engineering, Math, or a related field
- Deep experience with at least one:
- ML compiler frameworks (MLIR, TVM, XLA, etc.)
- Low-level optimization (scheduling, memory, tiling)
- Proven ability to design non-trivial compiler systems or passes
- Strong intuition for performance across compute, memory, and data movement
- Comfort working with hardware constraints
Nice to have:
- Experience with constraint solvers (MILP, ILP, CP)
- Background in accelerator architectures or embedded systems
- Experience optimizing ML workloads for latency/power
- Familiarity with DSP or real-time signal processing
Benefits
- 401(k)
- Medical insurance
- Vision insurance
- Dental insurance
- Commuter benefits
- Disability insurance
- Paid maternity leave
- Paid paternity leave
- Child care support
femtoAI is an equal opportunity employer committed to a diverse workforce which strives to create an inclusive working environment empowering everyone to do their best work. We do not discriminate on the basis of race, ethnicity, religion, gender, gender identity, sexual orientation, age, marital status, veteran status, or disability status.
Similar Jobs
Automotive • Big Data • Information Technology • Robotics • Software • Transportation • Manufacturing
The Senior Compiler Engineer will develop a model compilation toolchain for deploying optimized machine learning models in autonomous vehicles, focusing on performance engineering and collaboration with cross-functional teams.
Top Skills:
C++CublasCudaCudnnJaxMlirOnnxPythonPyTorchTensorFlowTensorrtTvmXla
Artificial Intelligence • Computer Vision • Hardware • Robotics • Metaverse
Develop algorithms and optimizations for inference and compiler stack, enhance performance metrics, collaborate with hardware teams, and publish work on novel approaches.
Top Skills:
C/C++LlvmMlirOnnxPyTorchRustTensorFlow
.png)
eCommerce • Fintech • Hardware • Payments • Software • Financial Services
The Territory Account Executive will oversee sales in their area, engage local businesses, conduct demos, and close deals while building partnerships and generating leads.
Top Skills:
Salesforce
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
