US 2026 Software - Machine Learning Intern

TETRAMEM INC; San Jose (CA); On-site; USD 48,216 - 61,992; Full time

Job summary

TetraMem is offering a Software Intern role in the US to support the development of tools, frameworks, and applications that enable neural networks to run efficiently on our compute-in-memory chips. This internship sits at the intersection of software, hardware, and AI.

The candidate should be pursuing CS/EE, with strong Python and C++ skills and familiarity with ML frameworks. This is a US-based internship for 2026 with competitive hourly pay.

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.

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

Skills

Python

C++

Data structures

Algorithms

AI/ML frameworks

Education

CS/EE degree student

Tools

TensorFlow

PyTorch

Job description

At TetraMem, we are redefining the future of AI with our groundbreaking innovations in In-Memory Computing. Leveraging world-record multi-level RRAM technology, we deliver highly efficient solutions for AI computations, enabling superior performance and energy efficiency across applications ranging from edge devices to data centers. Our talented team of engineers and industry-leading executives drives this progress, making TetraMem a leader in advanced memory technologies. If you are passionate about cutting-edge technology and thrive in a fast-paced, collaborative environment, TetraMem is the place for you. Join our global team to shape the future of AI computations and sustainable technology solutions while working at the forefront of innovation. Together, we can make a lasting impact.

Are you ready for new challenges and new opportunities?

Join our team!

Current job opportunities are posted here as they become available.

Subscribe to our RSS feeds to receive instant updates as new positions become available.

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

TetraMem celebrates diversity and is committed to creating an inclusive environment for all employees. We are proud to be an Equal Opportunity Employer and welcome applicants from all backgrounds. Qualified candidates will receive consideration for employment without regard to race, color, religion, creed, sex, gender identity or expression, sexual orientation, national origin, ancestry, age, marital status, medical condition, disability, genetic information, military or veteran status, or any other characteristic protected by applicable federal, state, or local law.

TetraMem is committed to providing reasonable accommodations to qualified applicants with disabilities throughout the recruitment process. Applicants requiring accommodation may contact Human Resources for assistance.