ML Engineer - Inference

Phota Labs is building visual GenAI to enhance photography experiences, and they are seeking an ML Engineer specializing in inference and optimization. In this role, you will bridge the gap between research models and production systems, transforming PyTorch research code into optimized solutions for user-facing applications.

Responsibilities:

• 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
• 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