Machine Learning Engineer – PyTorch Image Modeling

Tower 33 is a full service VFX studio. We pride ourselves on our client driven workflows and have garnered most of our success from word of mouth. Our clients are our lifeblood and our passion is bridging the gap between your imagination and final execution. We customize our approach to each individual project in order to achieve maximum efficiency and value.

Job Description: We are seeking a talented and experienced Machine Learning Engineer to spearhead the creation of a machine learning tool set for use within a visual effects workflow.  These tools will be built around open-source machine learning models.  With the aid of industry-leading outside consultation, the developer should be able to incorporate existing diffuser and image to image models to achieve a variety of defined image modification tasks.

This toolset will be developed using the PyTorch framework.  By leveraging TorchScript, these tools will be directly usable within existing visual effects workflows to aid in compositing tasks.  Ultimately, we would like to take this toolset outside of existing platforms and develop a standalone application.  To this end, we are open to developing tools outside of PyTorch, should this route provide more benefit.

Responsibilities:

• Develop and maintain machine learning models using PyTorch.
• Train, test and deploy deep learning models in production.
• Work with large image and video datasets and perform data preprocessing and augmentation.
• Finetune open-source  image models for visual effects tasks.
• Collaborate with project manager(s) and outside consultation to develop strategies for machine learning solutions.
• Stay up-to-date with the latest advancements in machine learning and implement best practices.

Requirements:

• Bachelor’s or Master’s degree in Computer Science, Statistics, Mathematics or related field - or equivalent experiences.
• Strong experience with PyTorch.
• Experience with OpenCV, diffusion models, and GPU programming
• Strong programming skills in Python and familiarity with open source tools and libraries.
• Experience in data preprocessing and data augmentation techniques.
• Experience with large datasets and distributed computing.
• Knowledge of machine learning best practices and techniques.
• Excellent problem-solving skills and ability to work independently.

Preferred Qualifications:

• Experience with visual effects workflows, and the language surrounding them.
• Experience with GUIs - the ability to package these tools inside of a standalone, user friendly interface.
• Experience with computer vision or natural language processing.
• Experience with deep reinforcement learning.
• Experience using AWS EC2, lambda, Sagemaker or the Azure/GCP equivalent.

This is a new venture for Tower 33, and offers the opportunity for significant expansion and growth on the success of the toolset. If you are a passionate and driven Machine Learning Engineer with experience in PyTorch and are excited about joining a dynamic team, please apply by sending your resume and cover letter to production@tower33vfx.com