ML Research Engineer – Diffusion Image AI & GPU Training

EliteHunt; New York (NY); On-site; USD 150,000 - 230,000; Full time

Benefits offered by this job

Competitive salary

Equity

Benefits

Job summary

A technology company in New York is seeking a Machine Learning Research Engineer to develop innovative machine learning models for product photography. You will contribute across all project phases, utilizing skills in Python, CUDA, and frameworks like TensorFlow and PyTorch. The role requires 2 to 10 years of relevant experience, including multi-node GPU training. Competitive compensation, equity, and benefits are offered, along with the opportunity to work in a dynamic team aiming to reshape e-commerce photography.

Qualifications

• 2 to 10 years of experience in a machine learning intensive role or in a research capacity.
• Strong experience with Python, CUDA, and deep learning frameworks.
• Proven experience in implementing and deploying multi-node GPU training.

Responsibilities

• Contribute to all phases of machine learning projects from research to deployment.
• Implement and deploy multi-node GPU training.
• Develop and maintain robust machine learning models.
• Stay informed of advancements in machine learning research.
• Translate complex research papers into practical solutions.
• Communicate technical concepts effectively.

Skills

Python

CUDA

TensorFlow

PyTorch

Multi-node GPU training

Communication skills

Job description

A technology company in New York is seeking a Machine Learning Research Engineer to develop innovative machine learning models for product photography. You will contribute across all project phases, utilizing skills in Python, CUDA, and frameworks like TensorFlow and PyTorch. The role requires 2 to 10 years of relevant experience, including multi-node GPU training. Competitive compensation, equity, and benefits are offered, along with the opportunity to work in a dynamic team aiming to reshape e-commerce photography.