Machine Learning Engineer (CUDA)

1 Month 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. Responsibilities include optimizing models using CUDA, developing efficient algorithms and data structures for GPU computation, collaborating with research and engineering teams, staying current with GPU technology advancements, and ensuring model efficiency across various GPU architectures. The ideal candidate will have strong C++ and CUDA programming skills, experience with deep learning frameworks (PyTorch/TensorFlow), and a deep understanding of parallel computing and GPU architecture. The role requires expertise in generative models and excellent problem-solving skills.
Must have:
  • C++ and CUDA programming
  • Experience with PyTorch or TensorFlow
  • Understanding of parallel computing and GPU architecture
  • Expertise in generative models (3DVAE, video diffusion)
  • Problem-solving and debugging skills
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

Every matrix - Experienced CRM Data Scientist

Every matrix

United Kingdom (Hybrid)
6 Months ago
Windranger Labs - AI Researcher

Windranger Labs

Singapore (On-Site)
1 Month ago
Dolby Laboratories - Senior Computer Vision Researcher

Dolby Laboratories

Bengaluru, Karnataka, India (Hybrid)
6 Months ago
ByteDance - Research Scientist in Foundation Model, Speech Understanding - 2024 Start (PhD)

ByteDance

Seattle, Washington, United States (On-Site)
5 Months ago
The Walt Disney Company - Sr Machine Learning Engineer

The Walt Disney Company

Seattle, Washington, United States (On-Site)
4 Months ago
Zazz - Machine Learning Engineer

Zazz

(Remote)
1 Month ago
Mistplay - Data Scientist Intern (Summer 2025)

Mistplay

Toronto, Ontario, Canada (Hybrid)
2 Months ago
Postman - Software Engineer - Applied AI Engineer

Postman

Bengaluru, Karnataka, India (On-Site)
6 Months ago
ByteDance - Research Scientist Graduate (Foundation Model - Generative AI) - 2025 Start (PhD)

ByteDance

San Jose, California, United States (On-Site)
3 Months ago
Zoox - Software Engineer - 3D World Generation Pipelines

Zoox

Foster City, California, United States (Hybrid)
5 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Starkflow - Principal Full Stack Developer

Starkflow

Bengaluru, Karnataka, India (Hybrid)
2 Months ago
PwC - IN-Senior Associate_ML Engineer_Data and Analytics_Advisory_Bangalore

PwC

Bengaluru, Karnataka, India (On-Site)
5 Months ago
Meta - Software Engineer, Machine Learning

Meta

Redmond, Washington, United States (On-Site)
5 Months ago
Dolby Laboratories - Senior Foundational AI Researcher

Dolby Laboratories

Bengaluru, Karnataka, India (Hybrid)
6 Months ago
Google - Senior Software Engineer, Machine Learning, YouTube

Google

San Bruno, California, United States (On-Site)
3 Months ago
The Walt Disney Company - Lead Machine Learning Engineer

The Walt Disney Company

Washington, United States (On-Site)
2 Months ago
ByteDance - DevOps Engineer - Applied Machine Learning Engine (Singapore)

ByteDance

Singapore (On-Site)
4 Months ago
NVIDIA - Research Scientist, Digital Humans - New College Grad 2025

NVIDIA

Santa Clara, California, United States (On-Site)
2 Months ago
ByteDance - Video Analysis and Quality Algorithm Intern 2023 Summer/Fall (MS)

ByteDance

San Jose, California, United States (On-Site)
5 Months ago

Get notifed when new similar jobs are uploaded

Jobs in San Francisco, California, United States

Onward Search - B2B Sales Associate

Onward Search

Baltimore, Ohio, United States (On-Site)
5 Months ago
Epic Games - Senior Engine Programmer, Verse Framework

Epic Games

Cary, North Carolina, United States (On-Site)
2 Months ago
Probably Monsters - Lead Site Reliability Engineer

Probably Monsters

Texas, United States (On-Site)
1 Month ago
The Walt Disney Company - Sr Software Engineer

The Walt Disney Company

Santa Monica, California, United States (On-Site)
2 Months ago
Google - Software Engineer III, Google Cloud Platforms

Google

Kirkland, Washington, United States (On-Site)
5 Months ago
Riot Games - Staff Software Engineer, Full-Stack - 2XKO

Riot Games

Los Angeles, California, United States (On-Site)
6 Months ago
The Walt Disney Company - Media Engineer II

The Walt Disney Company

Charlotte, North Carolina, United States (On-Site)
4 Months ago
Rackspace Technology - Inventory Control Specialist I

Rackspace Technology

Richardson, Texas, United States (On-Site)
1 Month ago
Thatgamecompany - Engine UI Programmer

Thatgamecompany

United States (Remote)
7 Months ago
Hitachi - D365 F&O Finance Architect

Hitachi

Chicago, Illinois, United States (Remote)
5 Months ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Interface AI - Software Development Engineer IV - Backend

Interface AI

India (Remote)
1 Month ago
Hudl - Staff Engineer - Machine Learning

Hudl

London, England, United Kingdom (Hybrid)
4 Months ago
ByteDance - Research Scientist- Foundation Model, Video Generation

ByteDance

Seattle, Washington, United States (On-Site)
5 Months ago
ByteDance - Research Scientist- Foundation Model, Generative AI

ByteDance

San Jose, California, United States (On-Site)
5 Months ago
Alpha Sense - Lead AI Platform Engineer

Alpha Sense

New York, New York, United States (On-Site)
4 Months ago
Zoox - Staff Software Systems Engineer - Software Architecture

Zoox

Foster City, California, United States (Hybrid)
5 Months ago
Ubisoft - Senior ML Data Scientist

Ubisoft

Montreal, Quebec, Canada (On-Site)
2 Months ago
NVIDIA - Research Scientist, AI Accelerator SW/HW Codesign

NVIDIA

Santa Clara, California, United States (On-Site)
2 Months ago
ByteDance - Student Researcher (Doubao (Seed) - Foundation Model - Speech & Audio) - 2025 Start (PhD)

ByteDance

Seattle, Washington, United States (On-Site)
5 Months 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

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)

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