AI Software Architecture Intern - Remote

Overview

As an AI Software Architecture Intern at d-Matrix, you will contribute to the integration of accelerators into the PyTorch ecosystem, focusing on memory management techniques.

In Short

• Internship position at d-Matrix in Santa Clara, CA.
• Focus on AI software architecture and memory management.
• Work with C++, Python, and PyTorch.
• Contribute to enhancing PyTorch’s execution and representation.
• Engage in optimizing memory management techniques.
• Collaborate with a diverse and inclusive team.
• Gain experience in LLM inference and computing architecture.
• Work in a hybrid environment, onsite 3-5 days a week.
• Emphasize respect and collaboration in the workplace.
• Equal opportunity employer committed to inclusivity.

Requirements

• Experience with C++ and Python.
• Familiarity with PyTorch internals.
• Passion for tackling challenges in AI.
• Strong communication and collaborative skills.
• Ability to learn and adapt in a fast-paced environment.

Benefits

• Opportunity to work at the forefront of AI technology.
• Inclusive and supportive team culture.
• Hands-on experience in software and hardware innovation.
• Mentorship from experienced professionals.
• Potential for future career opportunities within the company.