ML Software Intern: AI Tools on Novel MEM Chips

TetraMem - Accelerate The World; San Jose (CA); On-site; USD 48,000 - 62,000; Full time

Job summary

TetraMem - Accelerate The World in San Jose, CA is seeking a Software Intern to help build tools, frameworks, and applications for neural network models running on analog compute-in-memory hardware.

You will write Python or C++ code, optimize performance, and collaborate with AI researchers and hardware engineers to validate system functionality. This internship emphasizes hands-on software development and ML deployment within a cutting-edge AI Hardware space.

Qualifications

• 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 ML model deployment and runtime environments.
• Analyze and adapt ML models to improve 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

Software Architecture

AI/ML Frameworks

Education

Pursuing Bachelor's degree in Computer Science or Electrical Engineering

Tools

TensorFlow

PyTorch

Job description

TetraMem - Accelerate The World in San Jose, CA is seeking a Software Intern to help build tools, frameworks, and applications for neural network models running on analog compute-in-memory hardware.

You will write Python or C++ code, optimize performance, and collaborate with AI researchers and hardware engineers to validate system functionality. This internship emphasizes hands-on software development and ML deployment within a cutting-edge AI Hardware space.