Principal AI Software Architect, FlexAI
Delivering universal AI compute
Be an early applicant
Delivering universal AI compute
21-100 employees
Be an early applicant
21-100 employees
Company mission
To empower builders everywhere to develop groundbreaking AI solutions with effortless access to universal AI compute
Role
Who you are
- We are looking for a high-level AI expert with 15+ years of experience in software/ systems development and 2-3 years of hands-on experience in AI solutions architecture
- 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
- Experience designing and optimizing AI workloads for different hardware architectures (GPUs, TPUs, AI accelerators, custom chips)
- Hands-on programming expertise in C++, Python, Go, or Rust, with a focus on low-level optimization and performance tuning
- Knowledge of AI model deployment strategies, including model quantization, pruning, and compilation techniques
- Strong leadership and collaboration skills, with experience mentoring senior engineers and working with cross-functional teams
Desirable
- 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 the job involves
- 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
- This is a key technical leadership role with significant influence on FlexAI’s AI software ecosystem
- 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)
- Collaborate with cross-functional teams to align software and hardware roadmaps, ensuring seamless integration between AI workloads and FlexAI’s infrastructure
- Evaluate and implement innovations in distributed AI computing, parallelization techniques, and high-performance AI model execution
- Guide engineering teams by setting best practices, conducting design reviews, and mentoring senior engineers
- Represent FlexAI externally in AI/ML communities, conferences, and technical discussions with partners and stakeholders
Application process
- We usually respond within a day
Our take
With a vision to contribute to the AI revolution, FlexAI delivers a layer of software intelligence, abstraction and orchestration into the AI compute stack. By rearchitecting existing architecture, it is creating accessible, universal AI compute to support the future of machine intelligence.
The idea is to combat the existing challenges with running AI workloads across diverse hardware. By doing this, FlexAI will enable AI developers to leverage a broader range of computing power without needing to modify their code.
FlexAI only came out of stealth in 2024, so we've yet to see much of what it has to offer, but its proposition has attracted a hefty seed investment. This will support the company as it moves forwards with the launch of its first product and towards its ultimate goal to empower AI acceleration.
Freddie
Company Specialist at Welcome to the Jungle
Company
Funding (1 round)
Apr 2024
$30m
SEED
Total funding: $30m
Company benefits
- 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
Company HQ
Vivienne, Paris, France
Leadership
Brijesh Tripathi
(CEO)Spent 4 years at Intel Corporation; first as VP and CTO of the Client Computing Group, then VP of AXG. Was VP pf Hardware Engineering at Zoox, co-founded SmartLivez and worked in Engineering at Tesla, Apple and NVIDIA.
Dali Kilani
(CTO)Angel Investor, who is also a Board Member at Epigene Labs and Go My Code. Previously they were an Advisor for ProsperUs, CTO at Lifen and Cloud/Devops Architect for The Boston Consulting Group.
Salary benchmarks
We don't have enough data yet to provide salary benchmarks for this role.
Submit your salary to help other candidates with crowdsourced salary estimates.
Share this job
View 10 more jobs at FlexAI