Principal AI Scientist, FlexAI
Model Optimization
+ Benefits package
Delivering universal AI compute
Be an early applicant
Delivering universal AI compute
21-100 employees
Be an early applicant
+ Benefits package
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 Principal AI Data Scientist with deep technical expertise in model optimization to spearhead the development of high-performance AI models derived from both open-source architectures and customer-proprietary models
- This role is ideal for a hands-on leader who thrives at the intersection of cutting-edge AI, systems engineering, and real-world scalability
- Master’s or Ph.D. in Computer Science, Machine Learning, or related technical field
- 8–10+ years in machine learning, AI R&D, or systems optimization roles
- Deep experience with model optimization techniques: quantization (INT8/FP16), pruning, distillation, NAS, etc
- Proven track record of working with LLMs, vision models, or multimodal architectures
- Experience with deep learning frameworks (e.g., PyTorch, TensorFlow) and optimization tools (e.g., ONNX, TVM, TensorRT, Hugging Face Optimum)
- Strong software engineering skills in Python and/or C++
- Demonstrated ability to translate AI research into high-performance production systems
- Excellent communication skills and a customer-first mindset
What the job involves
- You’ll lead initiatives focused on enhancing model efficiency, latency, and throughput across diverse hardware and deployment environments
- Model Development and Optimization:
- Take ownership of adapting, compressing, and optimizing open-source or proprietary models to meet specific performance goals—such as speed, accuracy, and resource efficiency—across various edge and cloud environments
- Tech Evaluation & Customization:
- Evaluate and benchmark open-source LLMs, CV, and multimodal models. Customize architectures to meet customer use-case requirements including quantization, pruning, distillation, and architecture search
- Hardware-Aware Optimization:
- Work closely with performance engineers to tailor models for GPUs, CPUs, and emerging accelerators (e.g., AMD, ARM, or edge chips)
- Customer Collaboration:
- Partner with strategic customers to understand application needs and co-develop custom model optimization workflows and pipelines
- Cross-Functional Leadership:
- Collaborate with product, engineering, MLOps, and systems teams to ensure end-to-end delivery of robust and production-grade models
- Innovation & Technical Strategy:
- Stay ahead of AI optimization trends, propose new approaches, and guide architectural decisions around tooling, frameworks, and methodology
Application process
- We usually respond within three days
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 12 more jobs at FlexAI