Role Overview: As a Machine Learning Engineer, you will design, implement, and manage the infrastructure and services powering groundbreaking models and video generation capabilities. Your responsibilities will encompass backend development, infrastructure management as code, and deploying cutting-edge machine learning models. Collaboration with a team of world-class engineers and researchers will be essential to creating disruptive technologies. This is a full-time, in-person role based in New York or San Francisco.
Key Responsibilities:
- Accelerate and deploy latent diffusion models, diffusion transformers, or state space models using PyTorch best practices and TensorRT.
- Build and scale cloud-based infrastructure in AWS and Oracle Cloud, leveraging Kubernetes (EKS) and Infrastructure as Code (Terraform).
- Develop and scale data pipelines and event streaming/processing systems using AWS tools like SQS and Kafka.
- Architect and implement REST APIs for distributed systems.
- Optimize deployment of machine learning models on NVIDIA hardware using Kubernetes and Docker.
Experience Required:
- Proven expertise in deploying and accelerating advanced machine learning models, particularly diffusion models.
- Hands-on experience with Kubernetes, Docker, and cloud infrastructure management.
- Proficiency with event-driven architectures and tools like SQS and Kafka.
- Minimum 5 years of professional experience in Python or Go.
- Startup experience is essential; the pace is fast, and a proactive mindset is critical.
Attributes and Traits:
- Self-starter with excellent time management and a results-oriented approach.
- Strong communication skills with the ability to collaborate effectively across teams.
- Mission-driven and deeply invested in building disruptive technologies.
- Practical understanding of operational costs and scaling infrastructure.
- Passionate about creating innovative solutions in a dynamic, challenging environment.
Benefits:
- Competitive compensation and equity.
- Comprehensive healthcare coverage (Silver PPO Medical, Vision, Dental).
- 401(k) plan.
- Office perks, including catered lunches and snacks.
