ML Engineer: Large-Scale Video Generation & Training

Socket.dev; California (MO); On-site; USD 180,000 - 240,000; Full time

Job summary

Socket.dev is seeking a machine learning engineer to advance video generation through pre-training, fine-tuning, and efficient inference. You will design training recipes, run large distributed jobs, and optimize models for production-scale deployment.

The role requires hands-on experience with large generative models, Python software engineering, and expertise in PyTorch or JAX within a fast-moving, research-driven environment.

Qualifications

• Bachelor's degree in Electrical Engineering, Computer Science, Computer Engineering, or relevant degree.
• Minimum of 3 years relevant industry experience.
• Experience with large-scale generative model training for video generation.
• Experience running distributed training across multi-node GPU clusters.
• Strong software engineering skills in Python, with PyTorch or JAX.

Responsibilities

• Train large generative models through pre-training, fine-tuning, and inference optimization.
• Design training recipes and run distributed training jobs.
• Collaborate with engineers and researchers to advance video generation products.

Skills

Python

Distributed training

PyTorch

JAX

Education

Bachelor's degree in EE/CS/CE

Tools

PyTorch

JAX

Job description

Socket.dev is seeking a machine learning engineer to advance video generation through pre-training, fine-tuning, and efficient inference. You will design training recipes, run large distributed jobs, and optimize models for production-scale deployment.

The role requires hands-on experience with large generative models, Python software engineering, and expertise in PyTorch or JAX within a fast-moving, research-driven environment.