Logo
Hedra

Machine Learning Engineer (CUDA)

Hedra, New York, New York, us, 10261

Save Job

Join to apply for the

Machine Learning Engineer (CUDA)

role at

Hedra Join to apply for the

Machine Learning Engineer (CUDA)

role at

Hedra This range is provided by Hedra. Your actual pay will be based on your skills and experience — talk with your recruiter to learn more. Base pay range $150,000.00/yr - $250,000.00/yr 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.

Read on to find out what you will need to succeed in this position, including skills, qualifications, and experience.

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. The Hedra team is fully in-person in SF/NY with a shared love of solving hard problems using whiteboards.

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.

Compensation Range: $150K - $250K

Seniority level

Seniority levelEntry level Employment type

Employment typeFull-time Job function

Job functionEngineering and Information Technology IndustriesTechnology, Information and Internet Referrals increase your chances of interviewing at Hedra by 2x Get notified about new Machine Learning Engineer jobs in

New York, NY . Brooklyn, NY $119,000.00-$155,000.00 5 days ago Software Engineer, Machine Learning (Multiple Levels) - Slack New York, NY $167,300.00-$334,600.00 4 days ago New York, NY $120,000.00-$180,000.00 2 weeks ago New York, NY $100,000.00-$250,000.00 1 month ago New York, NY $93,144.00-$133,063.00 6 days ago New York, NY $181,875.00-$200,645.00 1 week ago New York, NY $180,000.00-$290,000.00 3 months ago Machine Learning Engineer, Personalization New York, NY $136,878.00-$195,540.00 1 week ago New York, NY $150,000.00-$260,000.00 4 months ago New York, NY $160,000.00-$280,000.00 1 year ago New York, NY $150,000.00-$220,000.00 1 month ago New York, NY $141,000.00-$202,000.00 6 days ago New York, NY $187,000.00-$270,000.00 1 month ago New York, NY $70,000.00-$100,000.00 1 week ago New York, NY $138,250.00-$197,500.00 1 week ago New York, NY $150,000.00-$210,000.00 1 month ago New York, NY $150,000.00-$200,000.00 2 months ago Machine Learning Engineer, GenRecs, Personalization New York, NY $138,250.00-$197,500.00 1 week ago New York, NY $167,000.00-$185,500.00 1 week ago New York, NY $180,000.00-$240,000.00 3 months ago New York, NY $138,250.00-$197,500.00 6 days ago New York, NY $110,000.00-$250,000.00 3 months ago New York, NY $100,000.00-$300,000.00 2 weeks ago New York, NY $175,000.00-$250,000.00 4 weeks ago New York, NY $164,000.00-$198,000.00 1 week ago New York, NY $180,000.00-$280,000.00 7 months ago New York, NY $185,000.00-$195,000.00 2 months ago New York, NY $130,000.00-$180,000.00 6 months ago Machine Learning Scientist, NLP (All Levels) New York, NY $200,000.00-$300,000.00 2 months ago We’re unlocking community knowledge in a new way. Experts add insights directly into each article, started with the help of AI.

#J-18808-Ljbffr