Machine Learning Engineer (CUDA)

5 Days ago • All levels • Artificial Intelligence

Job Summary

Job Description

Hedra seeks a CUDA ML Engineer to optimize machine learning models (3DVAE and video diffusion models) for GPU performance using CUDA. Responsibilities include developing efficient algorithms and data structures for GPU computation, addressing performance bottlenecks in video generation, collaborating with research and engineering teams, staying updated on GPU technology, and ensuring efficient model execution across various GPU architectures. The ideal candidate possesses strong C++ and CUDA programming skills, experience with PyTorch or TensorFlow, and a deep understanding of parallel computing and GPU architecture. The role involves optimizing generative models for efficient training and inference and requires excellent problem-solving abilities.
Must have:
  • C++ and CUDA programming
  • Experience with PyTorch or TensorFlow
  • Understanding of parallel computing and GPU architecture
  • Optimize 3DVAE and video diffusion models for GPU performance
  • Develop efficient algorithms for GPU computation
Perks:
  • Competitive compensation and equity
  • 401k
  • Healthcare (Silver PPO Medical, Vision, Dental)
  • Lunch and snacks at the office

Job Details

Hedra is a pioneering generative media company backed by top investors at Index, A16Z, and Abstract Ventures. We're building Hedra Studio, a multimodal creation platform capable of control, emotion, and creative intelligence.

At the core of Hedra Studio is our Character-3 foundation model, the first omnimodal model in production. Character-3 jointly reasons across image, text, and audio for more intelligent video generation — it’s the next evolution of AI-driven content creation.

Note: At Hedra, we’re a team of hard-working, passionate individuals seeking to fundamentally change content and build a generational company together. You should have start-up experience and be a self-starter that is driven to build impactful products that change the status quo. You must be willing to work in-person in either NYC or SF.

Overview:

We are seeking a talented CUDA ML Engineer to optimize our machine learning models for high-performance computing on GPU hardware. The ideal candidate will have expertise in CUDA programming and a deep understanding of how to leverage GPU acceleration to maximize the efficiency of our 3DVAE and video diffusion models.

Responsibilities:

  • Optimize machine learning models, specifically 3DVAE and video diffusion models, for GPU performance using CUDA, ensuring efficient training and inference.

  • Develop and implement efficient algorithms and data structures for GPU computation, addressing performance bottlenecks in video generation tasks.

  • Work closely with the research and engineering teams to understand model requirements and performance bottlenecks, facilitating collaboration.

  • Stay current with the latest advancements in GPU technology and machine learning optimization techniques.

  • Ensure that our models run efficiently on various GPU architectures, supporting scalability for large-scale training.

Qualifications:

  • Bachelor’s degree in Computer Science, Electrical Engineering, or a related field, with a focus on high-performance computing.

  • Strong programming skills in C++ and CUDA, essential for GPU optimization.

  • Experience with deep learning frameworks that support GPU acceleration, such as PyTorch or TensorFlow, crucial for model implementation.

  • Understanding of parallel computing concepts and GPU architecture, given the need to optimize for hardware constraints.

  • Familiarity with machine learning models, particularly generative models, to align optimizations with model needs.

  • Excellent problem-solving and debugging skills, necessary for addressing performance issues.

Benefits:

  • Competitive compensation and equity

  • 401k (no match)

  • Healthcare (Silver PPO Medical, Vision, Dental)

  • Lunch and snacks at the office

We encourage you to apply even if you don't fully meet all the listed requirements; we value potential and diverse perspectives, and your unique skills could be a great asset to our team.

Similar Jobs

ByteDance - Architect - AML Engine

ByteDance

San Jose, California, United States (On-Site)
5 Months ago
Trend Micro - NLP / Prompt Engineer (VicOne_Automotive Security)

Trend Micro

Taipei City, Taiwan (On-Site)
6 Months ago
Meta - Software Engineer, Machine Learning

Meta

Los Angeles, California, United States (On-Site)
4 Months ago
VGW - Machine Learning Engineer

VGW

Perth, Western Australia, Australia (On-Site)
4 Weeks ago
PwC - Senior AI Developer - Roma [DIG]

PwC

Rome, Lazio, Italy (On-Site)
6 Months ago
NVIDIA - Research Scientist, Deep Learning and Computer Vision

NVIDIA

Hsinchu, Hsinchu City, Taiwan (On-Site)
1 Month ago
Scale AI - Software Engineer, GenAI Model Evaluation

Scale AI

San Francisco, California, United States (Hybrid)
5 Months ago
NVIDIA - Principal DGX Cloud Machine Learning Architect

NVIDIA

Canada (On-Site)
1 Month ago
Inworld AI - Product Manager (Technical)

Inworld AI

Mountain View, California, United States (On-Site)
5 Days ago
Lucid Reality Labs - ML/AI Engineer

Lucid Reality Labs

Poland (Remote)
6 Days ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Altagram Group - Data Science Internship/Workstudent

Altagram Group

Germany (On-Site)
5 Days ago
Ubisoft - Lead R&D Scientist

Ubisoft

Shanghai, Shanghai, China (On-Site)
6 Days ago
Zoox - Collision Avoidance System, Machine Learning Internship/Co-op

Zoox

Foster City, California, United States (On-Site)
5 Months ago
DraftKings - Director of Data Science

DraftKings

Boston, Massachusetts, United States (On-Site)
2 Weeks ago
Luxoft - Regular Data Engineer

Luxoft

(Remote)
4 Months ago
Match Group - Senior ML Software Engineering Team Leader

Match Group

Seoul, South Korea (Hybrid)
5 Months ago
Nintendo - Intern – Machine Learning Software Engineer (NTD)

Nintendo

Redmond, Washington, United States (On-Site)
4 Months ago
ByteDance - Tech Lead - Applied Machine Learning Algorithm

ByteDance

San Jose, California, United States (On-Site)
5 Months ago
PwC - Senior Data Scientist

PwC

Kuala Lumpur, Federal Territory Of Kuala Lumpur, Malaysia (On-Site)
6 Months ago
Canva - Senior Machine Learning Engineer - Specialist Platform and Experience

Canva

Surry Hills, New South Wales, Australia (Remote)
2 Weeks ago

Get notifed when new similar jobs are uploaded

Jobs in New York, New York, United States

The Walt Disney Company - Senior Product Manager I, Hulu Product (Web Experiences)

The Walt Disney Company

Seattle, Washington, United States (On-Site)
1 Week ago
Magnopus - Senior Interactive Producer

Magnopus

Los Angeles, California, United States (On-Site)
8 Months ago
Onward Search - 2D Graphic Designer

Onward Search

Chicago, Illinois, United States (Remote)
1 Month ago
Daybreak Game Company LLC - Software Development Engineer (Cardset)

Daybreak Game Company LLC

Renton, Washington, United States (Remote)
4 Months ago
Scopely - Vice President, Head of Internal Audit

Scopely

California, United States (On-Site)
1 Month ago
Onward Search - Technical Project Manager

Onward Search

Atlanta, Georgia, United States (Hybrid)
1 Month ago
ByteDance - Service Contract Insurance Business Solution and Operations - Global Payment

ByteDance

San Jose, California, United States (On-Site)
4 Months ago
The Walt Disney Company - ESPN Chinese/English Bilingual Marketing Intern, Summer 2025

The Walt Disney Company

Bristol, Connecticut, United States (On-Site)
1 Week ago
Interface AI - Senior Software Engineer, Backend

Interface AI

United States (Remote)
1 Month ago
Scientific Games  - Product Innovation & Development Specialist

Scientific Games

Alpharetta, Georgia, United States (On-Site)
1377 Years ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Microsoft - Member of Technical Staff - Software Engineer

Microsoft

Redmond, Washington, United States (On-Site)
1 Week ago
Mashgin - Senior Software Engineer, Computer Vision and Deep Learning

Mashgin

Palo Alto, California, United States (Hybrid)
5 Months ago
Google - Software Engineer III, Machine Learning, Google Cloud Compute Infrastructure

Google

Sunnyvale, California, United States (On-Site)
4 Months ago
NVIDIA - Senior Developer Relations Manager - Robotics

NVIDIA

Tokyo, Japan (On-Site)
2 Months ago
Electronic Arts - Senior Manager, Generative AI Software Engineering

Electronic Arts

Austin, Texas, United States (On-Site)
3 Weeks ago
Spell Brush - AI Infrastructure Engineer

Spell Brush

San Francisco, California, United States (On-Site)
5 Days ago
Lionbridge Games - Games Language AI Specialist (Linguist)

Lionbridge Games

Masovian Voivodeship, Poland (On-Site)
1 Week ago
Rackspace Technology - Practice Director - Data Engineering and AI/ML

Rackspace Technology

Texas, United States (Remote)
6 Days ago
Meta - Research Scientist Intern, Machine Perception for Input and Interaction (PhD)

Meta

Pittsburgh, Pennsylvania, United States (On-Site)
4 Months ago
Krafton  - Applied Research Scientist/Engineer - LLM Game Agent

Krafton

Seoul, South Korea (On-Site)
6 Days ago

Get notifed when new similar jobs are uploaded

About The Company

We are a creation lab building foundation models into products that power the next generation of human storytelling

New York, New York, United States (On-Site)

New York, New York, United States (On-Site)

New York, New York, United States (On-Site)

New York, New York, United States (On-Site)

New York, New York, United States (On-Site)

New York, United States (On-Site)

New York, New York, United States (On-Site)

New York, New York, United States (On-Site)

New York, New York, United States (On-Site)

New York, New York, United States (On-Site)

View All Jobs

Get notified when new jobs are added by Hedra

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug