Machine Learning Engineer (CUDA)

6 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. Responsibilities include leveraging CUDA for efficient training and inference, developing efficient algorithms and data structures for GPU computation, collaborating with research and engineering teams, staying updated on GPU technology and optimization techniques, 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 or TensorFlow), and a deep understanding of parallel computing and GPU architecture. They must also possess excellent problem-solving and debugging skills.
Must have:
  • C++ and CUDA programming
  • Experience with PyTorch or TensorFlow
  • Deep learning model optimization
  • Understanding of parallel computing
  • GPU architecture expertise
  • Problem-solving & 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

NVIDIA - Senior Manager, Internal GPU and HPC Computing Clusters

NVIDIA

Seattle, Washington, United States (On-Site)
5 Days ago
Meta - Research Scientist Intern, Smart Glasses in Wearables AI (PhD)

Meta

Menlo Park, California, United States (On-Site)
4 Months ago
ByteDance - Applied Scientist Intern (Computational Modeling & Optimization)

ByteDance

San Jose, California, United States (On-Site)
1 Week ago
NVIDIA - AI Computing Software Development Engineer, TensorRT

NVIDIA

Shanghai, Shanghai, China (On-Site)
2 Months ago
Dream Sports - Lead ML Scientist

Dream Sports

Mumbai, Maharashtra, India (On-Site)
7 Months ago
Ubisoft - Senior C++ Programmer - Machine Learning

Ubisoft

Montreal, Quebec, Canada (On-Site)
2 Weeks ago
CharacterAI - Software Engineer, Machine Learning Infrastructure

CharacterAI

New York, New York, United States (On-Site)
6 Days ago
ByteDance - Research Engineer Graduate (Vision AI Platform)

ByteDance

Seattle, Washington, United States (On-Site)
1 Month ago
Zoox - Principal Machine Learning Engineer

Zoox

Foster City, California, United States (On-Site)
5 Months ago
NVIDIA - Product Marketing MBA Intern, DGX Cloud - Summer 2025

NVIDIA

Santa Clara, California, United States (On-Site)
2 Weeks ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Meta - Software Engineer, Machine Learning

Meta

Burlingame, California, United States (On-Site)
4 Months ago
Mistplay - Data Scientist Intern (Summer 2025)

Mistplay

Toronto, Ontario, Canada (Hybrid)
1 Month ago
Virtuos - Machine Learning Engineer

Virtuos

Singapore (On-Site)
1 Week ago
ByteDance - Algorithm Engineer - Audio Understanding - Start 2025

ByteDance

Singapore (On-Site)
5 Months ago
ByteDance - Senior Software Engineer - Generative AI

ByteDance

San Jose, California, United States (On-Site)
2 Months ago
Thumbtack - Staff Software Engineer,  Machine Learning Infrastructure

Thumbtack

United States (Remote)
4 Months ago
Netflix - Research Scientist (L6) - Identity Algorithms

Netflix

Los Gatos, California, United States (On-Site)
5 Months ago
Canva - Senior Computer Vision Engineer - Photo AI

Canva

London, England, United Kingdom (Remote)
2 Months ago
NVIDIA - Solution Architect, Generative AI - Digital Human

NVIDIA

Canada (On-Site)
1 Month ago
Meta - Software Engineer, Machine Learning

Meta

San Francisco, California, United States (On-Site)
4 Months ago

Get notifed when new similar jobs are uploaded

Jobs in San Francisco, California, United States

Netflix - Engineering Manager, Page Scale & Performance Team

Netflix

United States (Remote)
1 Month ago
Oculus VR - VFX Artist

Oculus VR

Burlingame, California, United States (Remote)
1 Week ago
Super - Senior Corporate Accountant

Super

United States (Remote)
6 Days ago
31st Union - Expert Core Engineer

31st Union

San Mateo, California, United States (On-Site)
2 Months ago
Riot Games - Principal Software Engineer - VALORANT, Foundations, Build Platforms

Riot Games

Los Angeles, California, United States (On-Site)
6 Months ago
Zoox - Software Engineer - Simulation Traffic & Behavior Modeling

Zoox

Seattle, Washington, United States (Hybrid)
5 Months ago
On Location - Accounts Payable Associate - FIFA World Cup 26™

On Location

New York, New York, United States (On-Site)
1 Month ago
PlayStation Global - Staff Software Engineer - Android/UI

PlayStation Global

Carlsbad, California, United States (On-Site)
1 Week ago
NVIDIA - Senior Technical Program Manager – Silicon Solutions

NVIDIA

Santa Clara, California, United States (Hybrid)
1 Month ago
Universal Music - Senior Manager, Controls Assurance

Universal Music

California, United States (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

NVIDIA - Director, Regional Developer Relations - ROAP

NVIDIA

Singapore (Remote)
1 Month ago
Velotio Technologies - Data Scientist

Velotio Technologies

Maharashtra, India (Remote)
2 Weeks ago
ByteDance - Research Scientist in Foundation Model, Music Core Machine Learning Graduates - 2024 Start (PhD)

ByteDance

San Jose, California, United States (On-Site)
4 Months ago
Meta - Research Scientist Intern, Machine Perception for Input and Interaction (PhD)

Meta

Pittsburgh, Pennsylvania, United States (On-Site)
4 Months ago
Google - Senior Software Engineer, Core Machine Learning, Google Cloud

Google

Sunnyvale, California, United States (On-Site)
4 Months ago
Vigaet - Internship -AI Agents

Vigaet

(Remote)
4 Months ago
NVIDIA - AI Algorithm Engineer - Silicon Solution Group

NVIDIA

Shanghai, Shanghai, China (On-Site)
3 Weeks ago
Microsoft - Member of Technical Staff, AI - Reinforcement Systems

Microsoft

London, England, United Kingdom (On-Site)
1 Week ago
AI Fund - Curriculum Developer

AI Fund

Germany (Remote)
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

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)

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