Machine Learning Engineer (CUDA)

4 Months ago • All levels • Research Development

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

Avalanche Studios Group - Senior Rendering Programmer

Avalanche Studios Group

Stockholm, Stockholm County, Sweden (On-Site)
4 Months ago
505 Games - Senior Software Engineer - Technical Business Unit

505 Games

Milan, Lombardy, Italy (Remote)
1 Year ago
MixMob - Game Producer

MixMob

Vancouver, British Columbia, Canada (Remote)
1 Year ago
London stock Exchange - Research Intern (Vietnamese)

London stock Exchange

Penang, Malaysia (On-Site)
3 Months ago
bytedance - Research Scientist Graduate, (AI-Native database systems)- 2026 Start (PhD)

bytedance

Seattle, Washington, United States (On-Site)
1 Week ago
bytedance - Research Scientist - Multimedia Lab

bytedance

San Diego, California, United States (On-Site)
4 Months ago
Ubisoft - Principal R&D Scientist on Bots & Behaviors

Ubisoft

Bordeaux, Nouvelle-Aquitaine, France (Hybrid)
6 Months ago
Game freak - R&D Programmer: Battle System [Junior Class]

Game freak

Chiyoda City, Tokyo, Japan (On-Site)
2 Months ago
Pulse Point - Sr. Machine Learning Engineer, AdTech

Pulse Point

United Kingdom (Remote)
4 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Lulalend - Direct Marketing Specialist

Lulalend

Cape Town, Western Cape, South Africa (On-Site)
2 Months ago
BioFire - Manufacturing Engineer II

BioFire

Salt Lake City, Utah, United States (On-Site)
2 Weeks ago
zeta - Sr. Site Reliability Engineer

zeta

Bengaluru, Karnataka, India (On-Site)
9 Months ago
bytedance - Destination Service Manager - EMEA

bytedance

London, England, United Kingdom (On-Site)
4 Months ago
The Walt Disney Company - Senior Electrical Engineer

The Walt Disney Company

Lake Buena Vista, Florida, United States (On-Site)
3 Weeks ago
cirrus logic - SQA Test Engineer

cirrus logic

Taipei City, Taiwan (Hybrid)
2 Months ago
Reltio - Associate Customer Engineer

Reltio

Austin, Texas, United States (On-Site)
3 Weeks ago
Aledade - Senior Security (DevSecOps) Engineer II

Aledade

United States (Remote)
9 Months ago
Two Circles - Technical Consultant, Martech

Two Circles

London, England, United Kingdom (Hybrid)
1 Week ago
Lilith games - Data Operations Manager

Lilith games

Shanghai, China (On-Site)
1 Week ago

Get notifed when new similar jobs are uploaded

Jobs in New York, New York, United States

Crunchyroll - Senior Director, Global Theatrical Marketing

Crunchyroll

Los Angeles, California, United States (Hybrid)
2 Months ago
Qualcomm - Staff GPU Compiler Engineer

Qualcomm

San Diego, California, United States (On-Site)
1 Month ago
ChainGuard - Enterprise Business Development Representative - Southern California

ChainGuard

United States (Remote)
2 Weeks ago
saebo  - Gravity Compensation

saebo

United States (On-Site)
3 Weeks ago
Poppulo - Senior Customer Success Engineer

Poppulo

Denver, Colorado, United States (On-Site)
1 Month ago
Crowd Strick - Senior SDET - Windows Sensor

Crowd Strick

United States (Remote)
3 Weeks ago
Saviynt - Director, Technical Partner Enablement - IAM/IGA Practice

Saviynt

Atlanta, Georgia, United States (Remote)
6 Months ago
bytedance - Research Scientist, Applied GAI-Vision

bytedance

San Jose, California, United States (On-Site)
9 Months ago
Sony Interactive Entertainment - Character Artist

Sony Interactive Entertainment

Burbank, California, United States (On-Site)
2 Weeks ago
Adyen - Operations Manager, Commercial

Adyen

New York, United States (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Research Development Jobs

Accenture - AI / ML Engineer

Accenture

Pune, Maharashtra, India (On-Site)
3 Weeks ago
2K - Staff Applied Scientist - Economy

2K

London, England, United Kingdom (Hybrid)
1 Week ago
Google - Technical Program Manager III, AI/ML, Cloud AI Systems

Google

Austin, Texas, United States (On-Site)
3 Months ago
Demandbase - Senior Applied Scientist

Demandbase

San Francisco, California, United States (On-Site)
3 Weeks ago
Yahoo - Senior Research Scientist - Phish and Spam Detection

Yahoo

United States (Hybrid)
2 Months ago
Thales - Innovation and Applied-AI Engineer

Thales

Singapore (Hybrid)
2 Weeks ago
EMA - Front End Engineer - Agentic AI Experiences

EMA

Bengaluru, Karnataka, India (Hybrid)
1 Week ago
Reddit - Senior Software Engineer, AI Enablement

Reddit

Toronto, Ontario, Canada (Remote)
2 Months ago
CD PROJEKT RED - Researcher

CD PROJEKT RED

Warsaw, Masovian Voivodeship, Poland (Hybrid)
1 Week ago
CD PROJEKT RED - Lead AI Engineer

CD PROJEKT RED

Boston, Massachusetts, United States (Hybrid)
1 Month ago

Get notifed when new similar jobs are uploaded

About The Company

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 ourCharacter-3foundation 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.

San Francisco, California, United States (On-Site)

San Francisco, California, United States (On-Site)

San Francisco, California, United States (On-Site)

New York, United States (On-Site)

San Francisco, California, 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, 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