Software Engineer, Machine Learning

Summary

The job is for a Software Engineer - Machine Learning at Twelve Labs, a fast-moving AI company that recently raised $17M in seed funding. The role involves building and deploying machine learning pipelines, implementing best MLOps practices, and minimizing the time from model training to deployment on their ML infrastructure.

Requirements

• 5+ years of software development experience, including experience in deploying machine learning models
• 3+ years of experience in building and deploying an end-to-end machine learning pipeline, or equivalent
• Experience in establishing and maintaining secure software and system development environments
• Experience designing control and sandboxing systems for AI research
• Willingness to learn the emerging AI technology and a practical mindset on productization
• Black-box level of understanding in Transformer-based neural network
• Experience in system development in model serving and inference

Responsibilities

• Be responsible for Model Serving and ModelOps: manage model-related metadata (using the model registry), implement hardware-accelerated optimization for each model engine, and containerize models for efficient serving
• Construct an ML pipeline that proficiently serves the trained foundation models in our ML Infrastructure
• Implement best model validation practices by conducting automatic evaluation benchmarking and performing output comparisons
• Develop an automatic training/finetune pipeline that includes rigorous data and model validation against the baseline model

Preferred Qualifications

• Experience with MLOps for managing the entire machine learning lifecycle, including model registry and versioning functionalities
• Experience with hardware-accelerated optimization techniques
• Experience in identifying and mitigating pain-points in ML research & modeling processes
• ML research experience would be helpful, as this role requires interchangeable effort on both research side and software side

Benefits

• Full health, dental, and vision benefits
• Extremely flexible PTO and parental leave policy. Office closed the week of Christmas and New Years
• Remote-flexible, offices in San Francisco and Seoul and coworking stipend
• VISA support (such as H1B and OPT transfer for US employees)
• An open and inclusive culture and work environment
• Work closely with a collaborative, mission-driven team on cutting-edge AI technology