Machine Learning Engineer

Description

Join our research-oriented engineering team, and you'll build state-of-the-art generative models and applications, partner with cross-functional teams, and deliver end-to-end features to power the next-generation creative tools. The ideal candidate should have deep experience in generative modeling, care about long-term sustainable software development, and can drive features from concept all the way to delivery. This position requires a self-motivated individual with excellent interpersonal skills to effectively collaborate with all levels of the organization.

Minimum Qualifications

• MS + 2 years of industry experience or PhD in Computer Science, Computer Vision, Machine Learning, or related field with publications in generative models
• Hands-on experience training generative models such as diffusion models, GANs, VAEs, or autoregressive image models.
• Strong programming skills in PyTorch or JAX.

Preferred Qualifications

• Experience with compositional or layout-conditioned image generation, or image decomposition/editing
• Experience deploying models to memory or compute constrained environments.
• Experience with large-scale model training and parallelization.
• Track record of publications in top-tier machine learning or computer vision conferences (e.g., CVPR, ICCV, ECCV, NeurIPS/ICML, SIGGRAPH, CHI).