Machine Learning Engineer, Hedra

CUDA

Salary not provided

+ Equity

Tensorflow

C++

CUDA

PyTorch

Junior, Mid and Senior level

New York

San Francisco Bay Area

AI lab empowering creation with AI

Job no longer available

AI lab empowering creation with AI

21-100 employees

B2BArtificial IntelligenceContentDigital MediaSaaS

Job no longer available

Salary not provided

+ Equity

Tensorflow

C++

CUDA

PyTorch

Junior, Mid and Senior level

New York

San Francisco Bay Area

21-100 employees

B2BArtificial IntelligenceContentDigital MediaSaaS

Company mission

To build foundation models into products that power the next generation of human storytelling.

Job

Company

Role

Who you are

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

What the job involves

We are seeking a talented CUDA ML Engineer to optimize our machine learning models for high-performance computing on GPU hardware
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

Our take

The relationship between machine learning and online content has grown closer in recent years, and Hedra is a video content platform that is pushing this symbiosis into new territory. Working as a content engine, Hedra's online creation studio helps users generate dialogue, audio and video from words.

As a digital presence automation tool, Hedra is moving beyond the likes of DallE and generative AI video content into more nuanced territory. For example, its product known as Character is a video foundation model for longer-form video generation focusing on customizable AI-generated video characters.

Between the version one release of Character in June 2024 and the company's fundraise that same August, more than 1.6 million videos had already been generated via the platform. This demonstrates the demand that exists for Hedra's offering. A new fundraise in 2025, which reportedly takes the company’s valuation to $200 million, will support the advancement of the quality and controllability of its models.

Kirsty

Company Specialist at Welcome to the Jungle

Company

Funding (2 rounds)

May 2025

$32m

SERIES A

Aug 2024

$10m

SEED

Total funding: $42m

Company benefits

Equity
401k (no match)
Healthcare (Silver PPO Medical, Vision, Dental)
Gym access (at our office)

Company HQ

SoMa, San Francisco, CA

Leadership

Michael Lingelbach

(Founder & CEO)

Previously Regina Casper Graduate Research Fellow at Stanford University, was an Intern at NVIDIA, Postbaccaulaureate researcher at Salk Institute for Biological Studies and a Scholar at the Janelia Research Campus.

Salary benchmarks

We don't have enough data yet to provide salary benchmarks for this role.

Submit your salary to help other candidates with crowdsourced salary estimates.

Share this job

View 5 more jobs at Hedra