Photalabs Com

ML Engineer - Inference

San Jose, USA

H1B sponsorship availableDetected 312 days ago

Cloud PlatformsMachine LearningDeep LearningPyTorchLogisticsPhotographyResearch

About the role

As our first ML Engineer specializing in inference and optimization, you'll bridge the gap between cutting-edge research models and production systems.
Your expertise will transform PyTorch research code into highly optimized, low-latency inference solutions that power our user-facing applications.

Implement neural network optimization techniques including quantization, pruning, and architectural refinements while preserving model accuracy.
Develop efficient training and fine-tuning strategies with optimal precision trade-offs and parallelism.
Build and maintain scalable multi-GPU inference solutions with sophisticated model parallelism and serving architectures.
Collaborate with the research team to ensure optimization integrate smoothly with model development workflows.

We offer generous health, dental, and vision coverage, unlimited PTO, paid parental leave, and relocation support as needed.
You'll work closely with our GenAI researchers, vision ML engineers, and backend team to deliver exceptional performance.
You may be a strong fit if you: Have experience deploying and optimizing deep learning models for production environments, particularly with multi-GPU inference and large-scale model serving.

At Phota Labs, we're building visual GenAI that helps people capture, express, and relive their memories - in ways that feel effortless, personal, and emotionally resonant.
Our core technology enables personalized image generation that faithfully reflects who you are and the moments you experienced.
Our first goal is to bring visual GenAI into everyday photography.
We're a small team of researchers, engineers, and designers who have always been at the forefront of how people capture, edit, and share images and videos.
We build with our hands and hearts.
We believe GenAI is the next shift for photography, and are seeking builders who share this vision - people like us, like you.
We're just getting started!
The role: As our first ML Engineer specializing in inference and optimization, you'll bridge the gap between cutting-edge research models and production systems.
Deploy and integrate researcher-trained model checkpoints into our cloud infrastructure and production pipelines.
Conduct thorough performance profiling and benchmarking to identify and eliminate computational bottlenecks.

This listing is sourced directly from Photalabs Com's careers page and normalized into a canonical job model.