ML Engineering Intern (2025 Winter ML infra and Model Optimization)

Genies • Los Angeles, CA

76d • $45 • Hybrid

About The Position

Genies is looking for a ML Infra Intern to join our R&D group this summer. Based either in our Los Angeles or San Mateo offices (Hybrid), you will work closely with a dedicated and talented team of technical artists, engineers and artists. Together, you will explore new concepts and technologies to further Genies' mission of empowering users to develop their own avatar ecosystems. We're looking for someone who is passionate about creating high-quality visuals and has the technical foundation to help us build the next wave of digital identity.

Requirements

• Currently enrolled in or a recent graduate of a PhD or Master degree program in Computer Science, Artificial Intelligence, Machine Learning, or a related field.
• Solid understanding of machine learning concepts and recent computer vision.
• Familiarity with programming languages such as Python and experience with machine learning libraries (e.g., PyTorch, TensorFlow, scikit-learn, Hugging Face).
• Ability to conduct research and prototype solutions based on academic literature.
• Strong problem-solving skills and the ability to work independently as well as in a team environment.
• Excellent communication skills, with the ability to present findings clearly and effectively.

Nice To Haves

• Experience with model deployment frameworks (e.g., TorchServe, TensorRT, ONNX Runtime, SageMaker, or Ray Serve).
• Familiarity with distributed training, mixed-precision training, or FSDP/DeepSpeed.
• Previous experience in generative AI models (diffusion models, GANs, multimodal models).

Responsibilities

• Research State-of-the-Art (SOTA) models, implement model architecture and training loops, and run training experiments.
• Deploy optimized models for image generation and 3D generation across different platforms.
• Apply model compression, quantization, pruning, and optimization techniques to improve inference efficiency and reduce latency.
• Evaluate and benchmark model performance to ensure accuracy, speed, and reliability.
• Collaborate cross-functionally and present your findings, insights, and results to the team and broader company.

Benefits

• Comprehensive health insurance for you and your family (Anthem + Kaiser Options Available), Dental and Vision Insurance
• Flexible paid time off, sick time, and paid company holidays, in addition to paid parental leave, bereavement leave, and jury duty leave for full-time employees
• Health & wellness support through programs such as monthly wellness reimbursement
• Working in a brand new, bright, open-environment and fun office space - there's even a slide!
• Choice of MacBook or windows laptop

What This Job Offers

• Career Level: Intern
• Industry: Professional, Scientific, and Technical Services
• Education Level: Master's degree
• Number of Employees: 101-250 employees