Machine Learning Lead
SweedSalary not specified
Full Time
Senior (5 to 8 years)
$200,000 – $300,000Compensation
Expert & Leadership (9+ years)Experience Level
Full TimeJob Type
UnknownVisa
Video AI, Artificial Intelligence, BiotechnologyIndustries
Requirements
The ideal candidate should possess proven experience deploying deep learning models on GPU-based infrastructure, including NVIDIA GPUs and CUDA, along with strong knowledge of containerization technologies like Docker and Kubernetes. Proficiency in Python and deep learning frameworks such as PyTorch or TensorFlow is essential, as is familiarity with model optimization techniques like quantization, pruning, and distillation. Furthermore, the candidate should have hands-on experience with profiling and optimizing model inference, including batching, concurrency, and hardware utilization.
Responsibilities
As the ML Engineering Lead, you will drive the technical vision for deploying large-scale multimodal diffusion models in production, overseeing and contributing to core ML pipelines from GPU-based inference to model optimization. You will collaborate with researchers to adapt generative models for real-world performance and reliability, develop high-performance GPU-based inference pipelines, and implement automated CI/CD pipelines for model deployment. Additionally, you will be responsible for designing and maintaining serving infrastructure, exploring cutting-edge GPU acceleration frameworks, and ensuring robust monitoring and alerting for critical production systems.
Skills
Machine Learning
ML Engineering
Multimodal Video Diffusion Models
Generative Models
Low-latency Inference
High-throughput Inference
Audio-video Generation
Diffusion Architectures
Temporal Modeling
AI
Captions
Video captioning and translation services
About Captions
Captions.ai enhances video content by providing captioning and translation services tailored for content creators, social media influencers, marketing agencies, and businesses. Their main offerings include automatic subtitle generation, translation into 28 languages, and video compression to improve performance. These tools simplify the video production process, allowing users to produce professional-quality videos with ease. Unlike many competitors, Captions.ai uses a freemium model, offering basic services for free while charging for advanced features, which helps attract a large user base and convert free users into paying customers. The company's goal is to make high-quality video content accessible to a wider audience, and recent funding will support their growth and product development.
New York City, New YorkHeadquarters
2021Year Founded
$82.7MTotal Funding
SERIES_CCompany Stage
Consumer Software, EntertainmentIndustries
51-200Employees
Benefits
Health Insurance
Dental Insurance
Vision Insurance
401(k) Retirement Plan
401(k) Company Match
Commuter Benefits
Wellness Program
Unlimited Paid Time Off
Flexible Work Hours
Risks
Increased competition from startups like Beeble AI could challenge Captions' market position.
Integration challenges from AlpacaML acquisition may delay product enhancements.
Rapid expansion may stretch resources, potentially affecting service quality.
Differentiation
Captions offers AI-powered video editing with automatic subtitle generation and language dubbing.
The platform supports video compression for optimized performance and accessibility.
Captions uses a freemium model to attract a wide user base and convert to paid plans.
Upsides
Captions secured $60 million in Series C funding, indicating strong investor confidence.
The acquisition of AlpacaML enhances Captions' creative tools with AI rendering capabilities.
Expansion to web and desktop platforms increases accessibility and user engagement.
Related Jobs
RemoteRemoteHead of ML
BiorenderSalary not specified
- Full Time
- Senior (5 to 8 years)
RemoteDirector Machine Learning
MotionalSalary not specified
- Full Time
- Expert & Leadership (9+ years)
RemoteLead Machine Learning & DevOps Engineer
NateraSalary not specified
- Full Time
- Senior (5 to 8 years), Expert & Leadership (9+ years)
RemotePrincipal ML Infrastructure Engineer
UpworkSalary not specified
- Full Time
- Expert & Leadership (9+ years)
RemoteMLOps Engineer
Trunk Tools$140,000 - $200,000/year
- Full Time
- Junior (1 to 2 years)
RemoteSenior AI Infrastructure Engineer
TetraScienceSalary not specified
- Full Time
- Senior (5 to 8 years), Expert & Leadership (9+ years)
RemoteSenior Research Engineer - Enterprise Products
NVIDIASalary not specified
- Full Time
- Senior (5 to 8 years)
RemoteDirector of AI Engineering
TalkiatrySalary not specified
- Full Time
- Expert & Leadership (9+ years)
RemoteEngineering Manager, Machine Learning
RunwaySalary not specified
- Full Time
- Junior (1 to 2 years), Senior (5 to 8 years)
RemoteGPU Solutions Engineer
VultrSalary not specified
- Full Time
- Junior (1 to 2 years)
RemoteSenior Solutions Architect, Gen AI
NVIDIASalary not specified
- Full Time
- Expert & Leadership (9+ years)
RemoteSenior Software Engineer, MLOps
Headspace$130,000 - $180,000/year
- Internship
- Mid-level (3 to 4 years), Senior (5 to 8 years)