AI Systems, Model Optimization

Unconventional AI; Mountain View (CA); On-site; USD 180,000 - 290,000; Full time

Benefits offered by this job

Health benefits

401k matching

Unlimited PTO

Complimentary meals

Job summary

Unconventional AI is seeking a Member of Technical Staff, AI Systems, Model Optimization, to drive the path from model architecture to silicon. You will develop training techniques, optimizations, and infrastructure to run AI models efficiently on novel compute substrates, linking model design to tapeouts.

You will implement energy benchmarks, partitioning strategies, and quantization schemes while collaborating with hardware teams to translate requirements into concrete specs for tapeouts.

Qualifications

• MS/PhD or equivalent in AI/CS/Physics/Electrical Engineering or Applied Math.
• Deep, practical understanding of the modern AI/ML stack and optimized compilation on GPUs.
• Map state-of-the-art AI models to system performance and apply efficiency techniques.

Responsibilities

• Develop energy benchmarks and performance models for compute, memory, and energy trade-offs.
• Partition and map AI models to hardware with advanced optimization strategies.
• Develop and apply QAT, noise-aware training, and sparsification to fit substrates.
• Develop and optimize kernels using CUDA, Triton, or CUTLASS and debug ML codebases.
• Collaborate with AI model architects and hardware teams for tapeouts.

Skills

AI/ML fundamentals

GPU optimization

Performance profiling

Cross-functional collaboration

Education

MS/PhD in AI/CS/Physics/EE/Applied Math

Tools

PyTorch

Torch.compile

DDP

FSDP

CUDA

Triton

CUTLASS

Job description

About Unconventional

Since 2022, AI has entered the mainstream, reshaping entire industries from education and software development to fundamental consumer behaviors. This revolution has created an unprecedented demand for computation - a demand that is now fundamentally limited by energy, not just in the datacenter, but at a global scale.

About Unconventional

Since 2022, AI has entered the mainstream, reshaping entire industries from education and software development to fundamental consumer behaviors. This revolution has created an unprecedented demand for computation - a demand that is now fundamentally limited by energy, not just in the datacenter, but at a global scale. At Unconventional, our mission is to solve this. We are rethinking computing from the ground up to build a new foundation for AI that is 1000x more efficient. We're doing this by exploiting the rich physics of semiconductors, mapping neural networks directly to the device physics rather than relying on layers of inefficient abstraction.

The Role

As a Member of Technical Staff, AI Systems, Model Optimization, you will develop the path from model architecture to physical silicon. You will develop the training techniques, optimization strategies, and infrastructure required to make AI models run efficiently on our novel compute substrates, closing the loop between model design and tapeout.

What You'll Do

• Energy Benchmarking & Performance Modeling: Develop rigorous performance models to evaluate compute, memory, and energy trade-offs. Track pareto-optimality across models and hardware configurations.
• Advanced Mapping & Partitioning: Drive the partitioning and mapping of complex AI models down to hardware. Apply and invent advanced optimization strategies from first principles, including custom quantization schemes, sparsity/pruning, and distillation to fit the physical constraints of our substrates.
• Hardware-Aware Training: Develop and apply Quantization-Aware Training (QAT), noise-aware training, and sparsification techniques to adapt models to the physical constraints of our analog compute substrates, including memory footprint, connectivity, precision, and noise.
• GPU Optimization & Kernel Development: Develop and optimize kernels using low-level programming models like CUDA, Triton, or CUTLASS. Profile and debug complex ML codebases to resolve performance bottlenecks (training and inference).
• Cross-Functional Collaboration: Act as a translator between AI model architects and hardware/infrastructure engineering teams, converting model requirements into concrete specifications and codifying learnings for tapeouts.

Minimum Qualifications

• Education: An MS/PhD or equivalent research/project experience in a quantitative field such as AI/Machine Learning, Computer Science, Physics, Electrical Engineering, or Applied Math.
• Experience: Deep, practical understanding of the modern AI/ML stack and optimized compilation and execution of algorithms on modern GPU systems. Proven experience in profiling, identifying, and resolving performance bottlenecks in complex ML codebases.
• Systems Fluency: Demonstrated ability to map state-of-the-art AI model architectures (e.g., Transformers, Mixture of Experts, diffusion models) to system performance implications and apply advanced efficiency techniques such as sparsity, quantization, and distillation.
• Software Development: Deep experience with PyTorch, including its internals, torch.compile, and distributed data parallel (DDP) / fully sharded data parallel (FSDP) libraries.

Preferred Qualifications (Nice To Have)

• Training Infrastructure: Experience with production-grade training frameworks (e.g., Megatron-LM, DeepSpeed) and distributed training at scale.
• Unconventional Co-Design: A forward-looking perspective on co-designing training systems for unconventional computing paradigms that map closely to the physics of underlying systems.
• Next-Gen Efficiency: Research or practical experience in advanced approximation/compression techniques beyond standard quantization, including noise-aware or physics-constrained training.

Why Join Us?

• The Mission: Redefine computing for the next 50 years by solving the fundamental energy limitation of AI at a global scale.
• The Impact: Shape the company's future as a foundational team member. Enjoy massive ownership and an outsized opportunity to drive change.
• The Perks: A comprehensive package including best-in-class health benefits, 401k matching, truly unlimited PTO, and complimentary meals in our Palo Alto office.