companyGeneralist logo

Software Engineer: ML Optimization

GeneralistSan Francisco Bay Area (San Mateo) or Boston (Somerville)
On-site Full-time

Clicking Apply Now takes you to AutoApply where you can tailor your resume and apply.


Unlock Your Potential

Generate Job-Optimized Resume

One Click And Our AI Optimizes Your Resume to Match The Job Description.

Is Your Resume Optimized For This Role?

Find Out If You're Highlighting The Right Skills And Fix What's Missing

Experience Level

Entry Level

Qualifications

The ideal candidate will have a strong foundation in machine learning and a keen interest in optimization techniques. A background in robotics or AI is a plus. Experience with CUDA, Python, and relevant ML frameworks is essential.

About the job

About the Role

Join our dynamic team, affectionately known as MBMB (More Big More Better), where you will play a crucial role in optimizing our training and on-robot inference stacks. We are seeking bold innovations that drive substantial improvements rather than incremental changes.

Your Responsibilities Will Include:

  • Maximizing GPU performance through innovative strategies
  • Deploying machine learning, hardware, and software modifications that yield significant advancements
  • Enhancing both inference and training stacks for optimal performance

Ideal Candidates Will:

  • Possess proficiency in the latest machine learning techniques, particularly for training and inference optimizations within transformer and diffusion-based architectures
  • Have a relentless pursuit of ML optimizations across various domains, including CUDA kernels, ML architecture, frontend and backend network bottlenecks, CPU inefficiencies, NVLink, and communication protocols, as well as optimizations in libraries such as Torch, NumPy, and Python.

About Generalist

At Generalist, we are committed to revolutionizing the landscape of general-purpose robotics. Our vision is to foster a future where humans and machines collaborate seamlessly to enhance productivity in various industries and daily life. We specialize in developing embodied foundation models with an emphasis on dexterity, pushing the boundaries of data, models, and hardware to empower robots to engage intelligently with the physical world. Our team comprises eminent researchers, roboticists, and innovators from leading organizations such as OpenAI, Boston Dynamics, and Google DeepMind, renowned for their groundbreaking contributions to AI. We have successfully launched and scaled transformative projects like ChatGPT and GPT-4, and we invite you to be part of this exciting journey.

Similar jobs

Tailoring 0 resumes

We'll move completed jobs to Ready to Apply automatically.