Genies is looking for an ML Infra Intern to join their R&D group to research, implement, and deploy state-of-the-art models for image and 3D generation, optimizing them for efficiency and performance across different platforms, contributing to the company's mission of powering the next era of interactive digital identity.
Requirements
- 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.
- 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.
Other
- 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.
- 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.
- Based either in our Los Angeles or San Mateo offices (Hybrid)
- This is a summer internship position.
