About the Company
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.
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.
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.
About the Role
Position Title: Principal AI Software Architect
Department: R&D SW
Location: Paris
Remote Status: Fully Remote
Employment Type: Full-time
Position Overview
FlexAI is seeking an exceptional Principal AI Software Architect to lead the design and development of our AI software stack, ensuring it is optimized for large-scale training and inference workloads. This role will drive technical strategy, define architectural frameworks, and collaborate closely with engineering and product teams to deliver world-class AI solutions.
Responsibilities
- Lead the architecture of FlexAI’s AI software stack, ensuring it supports scalable, efficient, and high-performance AI training and inference.
- Define and drive technical strategy for AI workloads, including model training, inference, and deployment across distributed environments.
- Optimize deep learning frameworks (e.g., PyTorch, TensorFlow) for performance, elasticity, and reliability at scale.
- Design and implement AI compute orchestration strategies that leverage a mix of cloud, on-prem, and specialized hardware (GPUs, TPUs, custom accelerators).
Qualifications
- 15+ years of experience in AI systems, distributed computing, or performance optimization for large-scale AI workloads.
- Proven expertise in AI solutions architecture, with at least 2-3 years leading AI software architecture efforts in a high-performance computing or cloud environment.
- Deep understanding of AI frameworks (e.g., PyTorch, TensorFlow, JAX) and their performance characteristics.
- Strong background in distributed computing and parallelization techniques, including multi-GPU training, HPC environments, and custom AI accelerators.
Bonus Points
- Experience in AI model parallelism techniques (e.g., tensor/model/data parallelism, pipeline parallelism).
- Knowledge of AI hardware/software co-design and experience collaborating with hardware teams on AI acceleration.
- Previous experience in startup environments or high-growth AI/ML-focused companies.
- Contributions to AI open-source projects, publications, or patents in AI computing.
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.
