US 2026 Software - Machine Learning Intern

About The Role

We are seeking a Software Intern to support the development of tools, frameworks, and applications that enable neural network models to run efficiently on TetraMem’s novel analog compute-in-memory chips. This is an exciting opportunity to work at the intersection of software, hardware, and artificial intelligence.

Responsibilities

• Develop and optimize Python or C++ code for neural network and machine learning model compression, conversion, deployment, and runtime environments
• Analyze and adapt ML models for to improve their compatibility and efficiency on compute-in-memory hardware and software
• Support software design, development, and performance profiling
• Collaborate with AI researchers and hardware engineers to validate system-level functionality
• Participate in code reviews, testing, and documentation

Qualifications

• Currently pursuing a degree in Computer Science, Electrical Engineering, or related field
• Solid programming experience in Python, C++, or similar
• Understanding of data structures, algorithms, and software architecture
• Familiarity with AI/ML frameworks (e.g., TensorFlow, PyTorch) is a plus
• Strong problem-solving skills and eagerness to learn

Salary Range: $35-45 USD/hr