What are the responsibilities and job description for the US 2026 Software - Machine Learning Intern position at TETRAMEM INC?
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 TetraMems 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
Salary : $35 - $45