Senior Machine Learning Engineer - Hardware Abstractions and Performance Optimization

Yesterday

Apply Now
Logo of Luma AI

Luma AI

Luma AI is a creative platform that enables users to ideate, visualize, and create stunning images and videos using advanced AI models without the need for complex prompt engineering. Featuring tools like Dream Machine, it allows users to direct videos, create unique characters, make easy edits, and explore endless creative ideas, all through natural language interactions. Luma AI is accessible on iOS and the web, making it a versatile tool for anyone looking to bring their creative visions to life.

Machine Learning β€’ Generative Media β€’ Generative AI β€’ AI Video

11 - 50 employees

Founded 2021

πŸ₯½ AR/VR

πŸ“± Media

πŸ“‹ Description

β€’ Luma’s mission is to build multimodal AI to expand human imagination and capabilities. β€’ Looking for engineers with significant experience maintaining & designing highly efficient systems and code that can be optimized to run on multiple hardware platforms. β€’ Ensure efficient implementation of models & systems. β€’ Identify and remedy efficiency bottlenecks by profiling and implementing high-performance PyTorch code. β€’ Benchmarking products across hardware & software to understand tradeoffs between latency, throughput and cost. β€’ Work closely with partners to identify bottlenecks. β€’ Collaborate with research team to ensure systems are planned efficiently and raise potential issues for hardware integration.

🎯 Requirements

β€’ Experience optimizing for memory, latency and throughput in Pytorch. β€’ Experience using torch.compile / torch.XLA. β€’ Experience benchmarking and profiling GPU & CPU code in Pytorch for optimal device utilization. β€’ Experience building tools & abstractions to ensure models run optimally on different hardware and software stacks. β€’ Experience working with transformer models and attention implementations. β€’ Experience with parallel inference, particularly with tensor parallelism, pipeline parallelism.

πŸ–οΈ Benefits

β€’ Offers Equity

Apply Now

2 days ago

Seeking a Demand Generation Lead to orchestrate campaigns for a B2B demand generation agency.

5 days ago

Wpromote

201 - 500

Join Wpromote as a Senior Paid Search Manager to oversee SEM campaigns for fintech clients. Support inclusive work culture while driving client success.

6 days ago

As Associate Director of Paid Media at TechnologyAdvice, lead teams to optimize paid media campaigns and revenue growth.

March 6

Drive and launch integrated campaigns to enhance sales efforts at PandaDoc, a leading document workflow platform.

Discover 100,000+ Remote Jobs!

Join now to unlock all job opportunities.

Find your dream remote job

Discover hidden jobs

We scan the internet everyday and find jobs not posted on LinkedIn or other job boards.

Head start against the competition

We find jobs within 24 hours of being posted, so you can apply before everyone else.

Be the first to know

Daily emails with new job openings straight to your inbox.

Choose your membership

Cancel anytime

Loved by 10,000+ remote workers

Wall of Love

Frequently asked questions

We use powerful scraping tech to scan the internet for thousands of remote jobs daily. It operates 24/7 and costs us to operate, so we charge for access to keep the site running.

Of course! You can cancel your subscription at any time with no hidden fees or penalties. Once canceled, you’ll still have access until the end of your current billing period.

Other job boards only have jobs from companies that pay to post. This means that you miss out on jobs from companies that don't want to pay. On the other hand, Remote Rocketship scrapes the internet for jobs and doesn't accept payments from companies. This means we have thousands more jobs!

New jobs are constantly being posted. We check each company website every day to ensure we have the most up-to-date job listings.

Yes! We’re always looking to expand our listings and appreciate any suggestions from our community. Just send an email to Lior@remoterocketship.com. I read every request.

Remote Rocketship is a solo project by me, Lior Neu-ner. I built this website for my wife when she was looking for a job! She was having a hard time finding remote jobs, so I decided to build her a tool that would search the internet for her.

Why I created Remote Rocketship

Choose your membership

Cancel anytime

Loved by 10,000+ remote workers
Built by Lior Neu-ner. I'd love to hear your feedback β€” Get in touch via DM or lior@remoterocketship.com