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

Genies is an avatar technology company powering the next era of interactive digital identity through AI companions. With the Avatar Framework and intuitive creation tools, Genies enables developers, talent, and creators to generate and deploy game-ready AI companions. The company’s technology stack supports full customization, AI-generated fashion and props, and seamless integration of user-generated content (UGC). Backed by investors including Bob Iger, Silver Lake, BOND, and NEA, Genies’ mission is to become the visual and interactive layer for the LLM-powered internet.

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. 

What You’ll be Doing: 

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

What You Should Have: 

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

Bonus Skills (Nice-to-Have)

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

Starting Salary: $45/hour

Here's why you'll love working at Genies:

• You'll work with a team that you’ll be able to learn from and grow with, including support for your own professional development
• You'll be at the helm of your own career, shaping it with your own innovative contributions to a nascent team and product with flexible hours and a work from home policy
• You'll enjoy the culture and perks of a startup, with the stability of being well funded 
• 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

Genies is an equal opportunity employer committed to promoting an inclusive work environment free of discrimination and harassment. We value diversity, inclusion, and aim to provide a sense of belonging for everyone.