Join FlexAI: 

FlexAI is at the forefront of revolutionizing AI computing by reengineering infrastructure at the system level. Our groundbreaking architecture, combined with sophisticated software intelligence, abstraction, and an orchestration layer, allows developers to leverage a diverse array of compute, resulting in efficient, more reliable computing at a fraction of the cost. We are seeking a skilled and experienced Senior ML System and Performance Engineer.

Founded by Brijesh Tripathi and Dali Kilani, who bring experience from Nvidia, Apple, Tesla, Intel, Lifen, and Zoox, FlexAI is not just building a product – we’re shaping the future of AI. Our teams are strategically distributed across Paris, Silicon Valley, and Bangalore, united by a shared mission: to deliver more compute with less complexity.

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:

Success at FlexAI requires an entrepreneurial spirit and startup mindset: the ability to rapidly iterate and make meaningful progress while staying focused on our mission to deliver more compute with less complexity. Your role is pivotal in realizing this goal by enabling systematic, reproducible benchmarking, comparison, and optimization of AI systems.

What you’ll need:

- Education: Master’s or Ph.D. in Systems Engineering, Artificial Intelligence, Machine Learning, or a related field. Ongoing professional development in AI/ML domains is expected.
- 7+ years of experience

- Proficiency in programming languages: Strong programming skills in Python and C++ with hands-on experience in PyTorch and other ML frameworks.

- GPU programming expertise: Strong knowledge of CUDA, ROCm, and related software ecosystems.

- Proficiency in AI models: Solid understanding of Transformer architectures and Generative AI (GenAI) models.

- Model and system optimization: In-depth knowledge of training, fine-tuning, and inference, with a focus on optimizing model and system performance.

- Proficiency in HPC libraries: Solid knowledge of computing libraries such as Triton, cuDNN, and cuBLAS to accelerate AI and machine learning workloads.

- MLOps expertise: Familiarity with modern MLOps techniques and toolsets for efficient model development and deployment pipelines.

What we offer:

• A competitive salary and benefits package, tailored to recognize your dedication and contributions.
• The opportunity to collaborate with leading experts in AI and cloud computing, learning from the best and the brightest, fostering continuous growth.
• An environment that values innovation, collaboration, and mutual respect.
• Support for personal and professional development, empowering you with the tools and resources to elevate your skills and leave a lasting impact.
• A pivotal role in the AI revolution, shaping the technologies that power the innovations of tomorrow.
• Hybrid position Paris based ( 3 days a week at the office)

Offices :

Our teams are strategically distributed across three continents—Europe, North America, and Asia—united by a shared mission: to deliver more compute with less complexity.

• Paris - HQ
• San Francisco (Bay Area) - US office
• Bangalore - India office

Apply NOW!

You’ve seen what this role entails. Now we want to hear from you. Does this opportunity align with your aspirations? If you’re even slightly curious, we encourage you to apply – it could be the start of something extraordinary!
We also kindly ask you to indicate in the notes section of your application if you are familiar with MLPerf benchmarks and to share your thoughts about them.

At FlexAI, we believe diverse teams are the most innovative teams. We’re committed to creating an inclusive environment where everyone feels valued, and we proudly offer equal opportunities regardless of gender, sexual orientation, origin, disabilities, veteran status, or any other facets of your identity that make you uniquely you.