Member of Technical Staff – Machine Learning Engineer

This is a fully remote position, open to applicants in United States.

📋 Description

• Transform innovative research into machine learning systems ready for production.

• Design, construct, and implement comprehensive ML models and pipelines.

• Develop and enhance models focused on image and video processing.

• Manage the entire ML lifecycle: experimentation, training/fine-tuning, evaluation, and deployment.

• Quickly create prototypes using open-source models and modify them to meet product requirements.

• Execute experiments, analyze outcomes, and iterate to enhance performance.

• Collaborate with researchers and interdisciplinary teams (product, engineering, design) to deliver scalable ML solutions.

• Engage with advancements in machine learning and apply them to perpetually enhance products.

⛳️ Requirements

• MS/PhD in Computer Science, Electrical Engineering, or a related discipline.

• Extensive research experience with knowledge of leading conferences (e.g., CVPR, ICCV, NeurIPS).

• Over 5 years of experience in Python and proficiency in Java, C++, or Scala.

• In-depth understanding of diffusion models.

• Strong grasp of multi-threading and memory management concepts.

• Solid understanding of ML architectures, including CNNs and Transformers.

• Proficiency with PyTorch or TensorFlow.

• Experience in building end-to-end ML deployment and inference systems, particularly for low-latency, real-time applications.

• Familiarity with deploying ML models in cloud environments (preferably AWS).

• Experience with experiment tracking systems and ML workflows.

🏝️ Benefits

• 5 weeks of paid leave to recharge.

• Comprehensive healthcare benefits, including vision and dental.

• Additional perks aimed at supporting your well-being.

• Visa assistance, including H1B and OPT transfers.