Stellon Labs Inc · 1 day ago
Research Engineer
San Francisco Bay Area
Full-time
Hybrid
Mid, Senior LevelStellon Labs Inc. is an AI research lab focused on making tiny frontier models for edge devices. The Research Engineer role involves developing novel architectures and methodologies for edge deployment, designing quantization techniques, and mentoring junior engineers while contributing to the research community.
Artificial Intelligence (AI)SaaSSoftware
Responsibilities
Develop novel architectures and training methodologies optimized for edge deployment scenarios
Design and implement advanced quantization techniques to compress neural networks while maintaining accuracy
Research and prototype new approaches to model compression, pruning, and knowledge distillation
Publish findings and contribute to the broader research community
Mentor junior engineers and drive technical excellence across the team
Qualification
Required
Develop novel architectures and training methodologies optimized for edge deployment scenarios
Design and implement advanced quantization techniques to compress neural networks while maintaining accuracy
Research and prototype new approaches to model compression, pruning, and knowledge distillation
Publish findings and contribute to the broader research community
Mentor junior engineers and drive technical excellence across the team
Quantization Expertise: Deep understanding of post-training quantization, quantization-aware training, and mixed-precision techniques
PyTorch Mastery: Extensive experience with PyTorch for research and production, including custom operators and model optimization
Model Training: Proven track record in training large-scale models, distributed training, and hyperparameter optimization
Master's or Ph.D. in CS, EE, Physics, or a related field (or equivalent experience) preferred
Publications in top-tier ML conferences/journals preferred
Preferred
Familiarity with training and inference of ultra-low precision models. Experience in writing CPU kernels would make one a strong fit
Background in writing CUDA(or similar) kernels
Experience with ONNX, TensorRT, or other model optimization frameworks
Knowledge of hardware accelerators (GPUs, TPUs, custom silicon)
Background in computer vision, NLP, or other specific ML domains
Open source contributions to ML optimization projects
Company
Stellon Labs Inc
Stellon Labs is an AI Research Lab that trains tiny frontier models that can run on edge devices.
Founded in 2025
2-10 employees
Funding
Current Stage
Early StageTotal Funding
$0.5MKey Investors
Y Combinator
2025-09-09Pre Seed· $0.5M
Company data provided by crunchbase