Machine Learning Software Engineering Intern

A company focused on unleashing the potential of generative AI through software and hardware innovation.
$62,400 - $104,000
Machine Learning
Software Engineering Intern
Hybrid
1 - 10 Employees
AI

Description For Machine Learning Software Engineering Intern

d-Matrix is at the forefront of generative AI innovation, combining software and hardware expertise to push technological boundaries. As a Machine Learning Software Engineering Intern, you'll join our R&D SW Infrastructure Software team in Santa Clara, working in a hybrid model (3 days onsite). The internship runs from June 2nd to August 15th, offering hands-on experience in developing cutting-edge ML models and AI infrastructure.

You'll work on implementing state-of-the-art models like LLaMA, GPT, and BERT, while also developing essential tools for performance simulation, analysis, and debugging. The role involves creating AI infrastructure software, including kernel compilers and inference engines, and contributing to QA systems development.

Our culture emphasizes respect, collaboration, and direct communication. We value diverse perspectives and seek passionate individuals ready to tackle complex challenges. The position offers competitive hourly compensation ($30-$50/hour) and valuable experience in a fast-paced startup environment.

Ideal candidates should be pursuing a degree in Computer Science or related field, with strong programming skills in Python/C/C++. Additional experience with ML frameworks, compilers, and hardware accelerators is beneficial. This role provides an excellent opportunity to work with exceptional professionals while contributing to the advancement of AI technology.

The position includes comprehensive benefits, including medical, dental, vision, and 401k, demonstrating our commitment to employee wellbeing. Join us in shaping the future of AI technology while growing your career in a supportive, innovative environment.

Last updated 5 days ago

Responsibilities For Machine Learning Software Engineering Intern

  • Develop performant implementations of SOTA ML models (LLaMA, GPT, BERT, DLRM)
  • Develop and maintain tools for performance simulation, analysis, debugging, profiling
  • Develop AI infra software including kernel compiler, inference engine, model factory
  • Develop QA systems/automation software
  • Collaborate with SW team to meet development milestones
  • Contribute to publication of papers and intellectual properties

Requirements For Machine Learning Software Engineering Intern

Python
  • Enrolled in a Bachelor's degree in Computer Science, Electrical and Computer Engineering, or related discipline
  • Problem-solving abilities to break down complex problems
  • Proficient in Python/C/C++ programming languages
  • Understanding of CPU / GPU architectures (desired)
  • Experience with specialized HW accelerators for deep neural networks (desired)
  • Experience with Machine Learning frameworks like PyTorch, Tensorflow (desired)
  • Experience with Machine Learning compilers like MLIR, TVM (desired)
  • Experience deploying inference pipelines (desired)

Benefits For Machine Learning Software Engineering Intern

Medical Insurance
Dental Insurance
Vision Insurance
401k
  • Medical Insurance
  • Dental Insurance
  • Vision Insurance
  • 401k
  • Comprehensive benefits package focused on employee wellbeing

Interested in this job?

Jobs Related To d-Matrix Machine Learning Software Engineering Intern

Research Scientist Intern, Computer Vision/Graphics/Machine Learning (PhD)

Research Scientist Internship at Meta focusing on Computer Vision, Graphics, and Machine Learning for AR/VR technologies, ideal for PhD students.

Software Engineer, Machine Learning

Software Engineering role focused on developing machine learning solutions at Meta

Software Engineer Intern, Summer 2025

Summer 2025 Software Engineering Internship at Empirical Health, building AI-powered healthcare solutions with wearable technology.

Software Engineer Internship, Machine Learning

Machine Learning Internship role at Runna, focusing on developing and fine-tuning ML models for personalized running training insights.

Graduate Software Engineer, Machine Learning (Graduating 2025)

Graduate Machine Learning Engineer position at Runna, building personalized training algorithms for runners using Python and AWS.