Outscal Logooutscal logo

Machine Learning Engineer (CUDA)

1 Hour 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

CharacterAI - Research Engineer - Multimodal

CharacterAI

Canada (On-Site)
6 Months ago
Netflix - Machine Learning Platform Product Manager, Training

Netflix

Los Gatos, California, United States (Hybrid)
2 Months ago
Tencent - Senior Researcher: Artificial General Intelligence (Natural Language Processing) 104531

Tencent

Bellevue, Washington, United States (On-Site)
2 Months ago
Rackspace Technology - Data Solutions Director

Rackspace Technology

United States (Hybrid)
3 Months ago
CharacterAI - Research Engineer - Multimodal

CharacterAI

Menlo Park, California, United States (On-Site)
7 Months ago
Rackspace Technology - Senior MLOPs Engineer (Canada)

Rackspace Technology

Toronto, Ontario, Canada (Remote)
4 Months ago
Netflix - Software Engineer L4/L5, Model Serving Systems, Machine Learning Platform

Netflix

Los Gatos, California, United States (Remote)
2 Months ago
ION - Senior AI Engineer, Italy

ION

Pisa, Tuscany, Italy (On-Site)
4 Months ago
NVIDIA - Solutions Architect, Hyperscale

NVIDIA

Santa Clara, California, United States (Hybrid)
1 Month ago
Netflix - Software Engineer L4/L5, Training Platform, Machine Learning Platform

Netflix

California, United States (Remote)
2 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

ThreeV Technologies,  Inc  - Data Scientist Computer Vision

ThreeV Technologies, Inc

Bengaluru, Karnataka, India (Remote)
4 Months ago
Rackspace Technology - Presales Data Science Architect – AWS Cloud

Rackspace Technology

Mexico City, Mexico (On-Site)
4 Months ago
Ubisoft - Senior ML Data Scientist

Ubisoft

Montreal, Quebec, Canada (On-Site)
1 Month ago
Takeda - Senior AI/ML Engineer

Takeda

Bengaluru, Karnataka, India (Hybrid)
4 Months ago
Riot Games - Senior Data Scientist - Singapore Efficiency Team

Riot Games

Singapore (On-Site)
1 Month ago
Alphasense - Join AlphaSense India Talent Community

Alphasense

Delhi, India (On-Site)
3 Months ago
Rackspace Technology - Data Scientist Intern

Rackspace Technology

Mexico City, Mexico City, Mexico (Remote)
2 Weeks ago
NVIDIA - Director, AI Software

NVIDIA

Taipei City, Taiwan (On-Site)
1 Month ago
ByteDance - Software Engineer Graduate (Applied Machine Learning - Engine) - 2025 Start (BS/MS)

ByteDance

San Jose, California, United States (On-Site)
4 Months ago
Epic Games - Chercheur scientifique principal

Epic Games

Montreal, Quebec, Canada (On-Site)
2 Months ago

Get notifed when new similar jobs are uploaded

Jobs in San Francisco, California, United States

Niantic - Visual & Motion Designer (Short-term)

Niantic

San Francisco, California, United States (Hybrid)
6 Months ago
Nintendo - Intern - Communications

Nintendo

Redmond, Washington, United States (On-Site)
4 Months ago
ByteDance - Student Researcher (Doubao (Seed) - Foundation Model - Vision and Language) - 2025 Start (PhD)

ByteDance

Seattle, Washington, United States (On-Site)
4 Months ago
ByteDance - Backend Software Engineer - Global E-Commerce Supply Chain Inventory

ByteDance

San Jose, California, United States (On-Site)
4 Months ago
King - Principal Technical Program Manager - Platform & Integrations | Activision Blizzard Media

King

San Francisco, California, United States (On-Site)
2 Months ago
My Fitness Pal - Product Marketing Manager

My Fitness Pal

United States (Remote)
1 Month ago
Inworld AI - Senior Software Development Engineer in Test (SDET) – Game Engine SDKs - USA

Inworld AI

Mountain View, California, United States (On-Site)
4 Months ago
The Walt Disney Company - Senior Digital Product Manager

The Walt Disney Company

Burbank, California, United States (On-Site)
3 Weeks ago
Onward Search - Inside Sales Representative

Onward Search

Louisville, Kentucky, United States (On-Site)
1 Month ago
Riot Games - Art Lead, Concept Art - VALORANT, Characters

Riot Games

Los Angeles, California, United States (On-Site)
2 Weeks ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Trend Micro - NLP / Prompt Engineer (VicOne_Automotive Security)

Trend Micro

Taipei City, Taiwan (On-Site)
5 Months ago
ByteDance - Senior Research Scientist, Foundation Model, Speech Understanding

ByteDance

Seattle, Washington, United States (On-Site)
4 Months ago
Zoox - Senior/Staff Software Engineer - Simulation Traffic & Behavior Modeling

Zoox

Foster City, California, United States (Hybrid)
4 Months ago
Zoox - Senior Software Engineer - Simulaton Scenario Automation

Zoox

Foster City, California, United States (Hybrid)
4 Months ago
Glean - Software Engineer, Machine Learning (Infrastructure)

Glean

Palo Alto, California, United States (On-Site)
3 Months ago
Trend Micro - NLP / Prompt Engineer (VicOne_Automotive Security)

Trend Micro

Taipei City, Taiwan (On-Site)
5 Months ago
HP - Machine Learning Engineer

HP

Palo Alto, California, United States (On-Site)
5 Months ago
HP - Machine Learning Intern

HP

Austin, Texas, United States (On-Site)
5 Months ago
Meta - Software Engineer, Systems ML - SW/HW Co-design

Meta

Sunnyvale, California, United States (Remote)
3 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)

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

California, United States (On-Site)

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

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

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

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

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

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

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