Machine Learning Engineer, Hedra

CUDA

Salary not provided

+ Equity

Tensorflow

C++

CUDA

PyTorch

Junior, Mid and Senior level

New York

San Francisco Bay Area

Creation lab building foundation models for next-gen human storytelling

Open for applications

Creation lab building foundation models for next-gen human storytelling

1-20 employees

B2BContentDigital MediaSaaS

Open for applications

Salary not provided

+ Equity

Tensorflow

C++

CUDA

PyTorch

Junior, Mid and Senior level

New York

San Francisco Bay Area

1-20 employees

B2BContentDigital 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. Describing itself as a video foundation mode company, Hedra is building an online creation studio that 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. Its most recent product is Character1, a video foundation model for longer-form video generation focusing on customizable AI-generated video characters.

Since Character1’s release in June 2024, more than 1.6 million videos have been generated via the platform, demonstrating the need for its August seed funding round. The $10 million investment will be channelled into developing new technology like a multimodal studio and workflow tool to complement its foundation model.

Kirsty

Company Specialist at Welcome to the Jungle

Company

Funding (1 round)

Aug 2024

$10m

SEED

Total funding: $10m

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 12 more jobs at Hedra