AI Compiler Engineer - Senior/Lead
Join us ! 🚀
🌎 Join FlexAI: Unleashing the Power of AI Through Purpose-Built Compute Systems
At FlexAI, we are on a mission to revolutionize the landscape of artificial intelligence by addressing the critical challenges in Large Language Model (LLM) training. Our motto "More compute, less complexity". With a commitment to sustainability, efficiency, and accessibility, we've gone back to the first principles of computer architecture to create a groundbreaking modular design. Our dedicated, cost-efficient, and open solutions for AI compute are engineered to deliver unparalleled performance per Total Cost of Ownership (TCO).
Led by two co-founders, with strong Silicon Valley DNA, bringing over 40 years of combined experience from industry giants (Intel, Apple, Tesla, Nvidia…), FlexAI is set to redefine the future of AI infrastructure.
If you're passionate about shaping the future of artificial intelligence, driving innovation, and contributing to a sustainable and inclusive AI ecosystem, FlexAI is the place for you !
🤖 Position Overview:
FlexAI is seeking a talented and innovative AI Compiler Engineer to join our cutting-edge team. In this role, you will be a key contributor to the design, development, and optimization of compilers tailored for FlexAI's purpose-built AI compute systems. As an AI Compiler Engineer, you will collaborate closely with hardware engineers, software developers, and system architects to ensure the seamless integration and optimal performance of compilers for AI workloads.
At FlexAI, we are on a mission to build a highly efficient and performant AI Stack for Model Training, aiming to positively disrupt the AI infrastructure space. To achieve this ambitious goal, we are seeking an outstanding AI Compiler Engineer, someone with a solid understanding of compiler optimizations and principles, machine learning and deep learning concepts, and a strong drive to make a significant impact.
Here's your canvas :
- Develop and optimize AI compiler techniques for improving the performance of deep learning/LLM models.
- Define and build the infrastructure to measure and track compiler performance in synthetic as well as functional benchmarks
- Collaborate with fellow engineers to design, develop, and debug AI models, tools, and libraries
- Contribute to the company's research and development efforts to drive AI/Machine Learning forward.
- Participate in performance analysis, tuning, and debugging session.
- Collaborate in a fast-paced team environment and meet demanding schedule goals.
You'll thrive with us if you have these elements :
- Bachelor's or higher degree in Computer Science, Electrical Engineering, or a related field.
- Proven experience as an AI Compiler Engineer, preferably in within the PyTorch ecosystem.
- Demonstrated experience in compiler optimizations and principles.
- Solid understanding of machine learning and deep learning concepts, Good knowledge/Experience with the PyTorch2 stack is a major plus
- Proficiency in programming languages such as C, C++, Python, Go or Rust, with a focus on compiler development.
- Experience with compiler optimization techniques and code generation for AI workloads.
- Experience with the MLIR infrastructure is highly desirable
- Knowledge of parallel computing architectures and vectorization.
- Strong analytical and problem-solving skills.
- Excellent collaboration and communication skills to work effectively with cross-functional teams.
- Excellent oral and written communication skills in English.
- Entrepreneur & start-up mindset
🙌🏼 What we offer:
- Competitive salary and benefits package, crafted to reward your efforts and commitment.
- A chance to work alongside pioneers in AI and cloud computing, learning from the best and brightest.
- An environment that values innovation, collaboration, and respect in every interaction.
- Opportunities for personal and professional growth, encouraging you to stretch your abilities and make a lasting impact.
- A platform to contribute significantly to the AI revolution, shaping the cloud services that power tomorrow's innovations.
🎯 Who is a Flexer?
At FlexAI, we're passionate about propelling AI forward while ensuring a culture of diversity and enjoyment. We cherish every individual's unique viewpoint, recognizing it as a catalyst for creativity and innovation. Committed to personal growth, we provide ongoing learning opportunities guided by our core values of ownership, optimism, and velocity. Join our dynamic, inclusive team, united by a common mission to achieve outstanding results while embracing the journey with joy.
Our Values :
⭐WE ARE (SERIOUSLY) OBSESSED BY OUR CUSTOMERS
We highly value our customers and are dedicated to meeting their needs. This commitment is reflected in our product development, customer service, and continuous feedback-seeking.
🌻 WE GET (BOLD) THINGS DONE
We prioritize productivity, efficiency, and proactive problem-solving. Encouraging innovative thinking, risk-taking, and goal-oriented execution.
🌔 WE RISE TO (BIG) CHALLENGES
We embrace challenges as opportunities for growth and learning. Our determination drives us to overcome obstacles and strive for excellence.
🦸♀️ WE ARE A (BADASS) TEAM
WE ARE A (BADASS) TEAM" symbolizes our unified strength, collaboration, and solidarity. Together, we bolster one another, exchange knowledge, and strive towards shared objectives. Each team member's contribution is valued, recognizing that our collective effort is paramount to our success.
🌝 WE CULTIVATE THE (SHARED) FUN
We foster a productive and enjoyable work environment. Fun promotes creativity, teamwork, and enhances employee satisfaction and retention.
♥️ WE EMBRACE (360°) DIVERSITY
We appreciate and validate diverse perspectives. Recognizing diversity as our strength, fueling creativity and innovation.
FlexAI strongly believes in the power of human ingenuity and AI reliability to redefine boundaries. We embrace a culture of experimentation, teamwork, equality, humble confidence, and hard work. If these values resonate with you, JOIN US !
📌 Offices :
FlexAI has European roots but international scope from day one. Our main offices are in
- Paris - HQ
- Silicon Valley - US office
- Bangalore - India office
- Remote friendly (Meet at the Paris office at least once per quarter 🍻)
🔍 Recruitment Process :
- HR interview (Recruiter) - 30 min (Phone or Video)
- #1 Technical Interview & Cultural fit (Dali, co-founder) - 45 min (Video)
- #2 Technical Interview (Tech Member) - 45 min (Video)
- #3 Technical Interview (Tech Member) - 45 min (Video)
- Team Fit Interview (2 Tech Members)- 30 min (Video)
- Final Interview (Brijesh, co-founder) - 45 min (Video)
- Referral check ✅
- Offer 🥳
- Day 1 & Onboarding 🚀
We've provided detailed insights into the role, and now it's your turn to assess if it aligns with your aspirations. If unsure, we encourage you to apply – it could lead to a meaningful collaboration.
We're committed to diversity and inclusivity, offering equal opportunities regardless of gender, sexual orientation, origin, disabilities, veteran status, or any other traits that make you who you are. Diversity enriches our workplace, making it more enjoyable for everyone !
- Department
- R&D SW
- Role
- AI Compiler Engineer
- Locations
- Europe - Remote, Paris, Bengaluru
- Remote status
- Hybrid
- Employment type
- Full-time
AI Compiler Engineer - Senior/Lead
Join us ! 🚀
Loading application form
Already working at FlexAI?
Let’s recruit together and find your next colleague.