Remote job
AI / ML Engineer
Job details
About this role
Role overview
A role for a machine learning engineer focused on building and refining frontier generative AI models across image, video, and audio modalities. The work centers on training at scale, optimization, and turning research advances into high-performing, deployable systems.
Responsibilities
- Design, train, and optimize large-scale generative models spanning image, video, and audio domains. - Develop and maintain training pipelines leveraging PyTorch and custom CUDA kernels. - Run experiments to improve model quality, training stability, and inference efficiency. - Profile, debug, and tune performance across GPU hardware and accelerators. - Collaborate with research and engineering peers to translate novel ideas into production-ready model systems.
Requirements
- Hands-on experience building models with PyTorch. - Proficiency with CUDA for GPU-accelerated training and inference. - Background in deep learning for generative models across at least one modality (image, video, or audio). - Comfort working on large-scale training runs and interpreting experimental results.