Machine Learning Engineer (CUDA)

2 Months 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 - Software Engineer - Applied Machine Learning, Engine

bytedance

San Jose, California, United States (On-Site)
1 Month ago
Qualcomm - Machine Learning & AI ADAS Applications Senior Staff Engineer

Qualcomm

Shanghai, China (On-Site)
2 Weeks ago
IMC - Machine Learning Engineer

IMC

Chicago, Illinois, United States (On-Site)
1 Month ago
Scale AI - Machine Learning Engineer, GenAI Applied ML

Scale AI

San Francisco, California, United States (On-Site)
1 Month ago
Tesla - Senior Machine Learning, AI Engineer

Tesla

Brandenburg, Germany (On-Site)
3 Months ago
bytedance - Research Engineer Intern (Doubao (Seed) - Machine Learning System) - 2025 Summer (MS)

bytedance

Seattle, Washington, United States (On-Site)
7 Months ago
Google - PhD Software Engineer

Google

Sunnyvale, California, United States (On-Site)
1 Month ago
bytedance - Backend Engineer (Model Inference), Machine Learning Systems

bytedance

Singapore (On-Site)
7 Months ago
Microsoft - Senior Applied Researcher

Microsoft

Redmond, Washington, United States (On-Site)
1 Month ago
Tencent - Senior Researcher, Natural Language Processing

Tencent

(On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Grab - Data Scientist (Dynamic Pricing)

Grab

Beijing, China (On-Site)
3 Weeks ago
bytedance - Research Scientist in Foundation Model, Speech Understanding - 2024 Start (PhD)

bytedance

San Jose, California, United States (On-Site)
7 Months ago
Intel  - GenAI Applications Intern

Intel

Guadalajara, Jalisco, Mexico (Hybrid)
3 Weeks ago
flip fit - Senior Machine Learning Engineer

flip fit

(Remote)
2 Months ago
bytedance - Research Scientist, Multimodality

bytedance

Seattle, Washington, United States (On-Site)
7 Months ago
Meta - Software Engineer, Machine Learning

Meta

Bellevue, Washington, United States (On-Site)
6 Months ago
Dream Sports - Lead ML Scientist

Dream Sports

Mumbai, Maharashtra, India (On-Site)
1 Month ago
bytedance - Research Engineer in Large Model System

bytedance

San Jose, California, United States (On-Site)
7 Months ago
The Walt Disney Company - Lead Software Engineer - Applied AI

The Walt Disney Company

Santa Monica, California, United States (On-Site)
1 Month ago
Microsoft - Principal Applied Science Manager

Microsoft

Bengaluru, Karnataka, India (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Jobs in New York, New York, United States

Visa - Visa Direct Product & Client Solutions Manager

Visa

Ashburn, Virginia, United States (Hybrid)
2 Weeks ago
Rockstar Games - Product Designer

Rockstar Games

New York, New York, United States (On-Site)
1 Month ago
Google - Senior Software Engineer, Security/Privacy

Google

Kirkland, Washington, United States (On-Site)
1 Month ago
Google - Software Engineer, PhD

Google

Sunnyvale, California, United States (On-Site)
1 Month ago
Treelix - Federal Sales Account Manager – Navy/USMC

Treelix

Reston, Virginia, United States (On-Site)
3 Weeks ago
CareerVillageorg - Volunteer: Scriptwriter

CareerVillageorg

United States (Remote)
9 Months ago
CrowdStricke - Sr. Taxonomy Specialist

CrowdStricke

United States (Remote)
3 Weeks ago
NVIDIA - Senior Site Reliability Engineer - Infrastructure

NVIDIA

Durham, North Carolina, United States (On-Site)
4 Months ago
Rockstar Games - Director of Security Operations

Rockstar Games

New York, New York, United States (On-Site)
2 Months ago
Playstation - Senior Data Scientist

Playstation

San Diego, California, United States (Hybrid)
4 Days ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Meta - Software Engineer, Machine Learning

Meta

Sunnyvale, California, United States (On-Site)
6 Months ago
ElevenLabs - Machine Learning Researcher

ElevenLabs

Poland (Remote)
2 Months ago
Inworld AI - Forward Deployed Engineer (AI Gameplay Engineer)

Inworld AI

Vancouver, British Columbia, Canada (On-Site)
2 Months ago
Talentica Software - Data Scientist

Talentica Software

India (Remote)
7 Months ago
CharacterAI - Software Engineer, Machine Learning Infrastructure

CharacterAI

New York, New York, United States (On-Site)
2 Months ago
Microsoft - Member of Technical Staff, AI - Reinforcement Learning Systems

Microsoft

Mountain View, California, United States (Hybrid)
1 Month ago
bytedance - Research Engineer - Multimodal Model

bytedance

Singapore (On-Site)
7 Months ago
Canva - Senior Machine Learning Engineer - Photo AI

Canva

Vienna, Vienna, Austria (Remote)
4 Months ago
NVIDIA - Principal Software Engineer - Enterprise AI Platform

NVIDIA

Santa Clara, California, United States (Hybrid)
4 Months ago
Assist software  - AI Engineer

Assist software

Suceava, Suceava County, Romania (Remote)
6 Months 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)

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