Machine Learning Engineer – PyTorch Image Modeling


Applications have closed

Tower 33 VFX

A collective of artists & full service VFX studio.

View company page

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.



  • 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.


  • 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


Tags: AWS Azure Computer Science Computer Vision Deep Learning Diffusion models EC2 GCP GPU Lambda Machine Learning Mathematics ML models NLP OpenCV Open Source Python PyTorch Reinforcement Learning SageMaker Statistics

Perks/benefits: Career development

Regions: Remote/Anywhere North America
Country: United States
Job stats:  1378  462  2

More jobs like this

Explore more AI, ML, Data Science career opportunities

Find even more open roles in Artificial Intelligence (AI), Machine Learning (ML), Natural Language Processing (NLP), Computer Vision (CV), Data Engineering, Data Analytics, Big Data, and Data Science in general - ordered by popularity of job title or skills, toolset and products used - below.