Software Intern: AI & Compute-In-Memory (Python/C++)

TETRAMEM INC; Singapore; On-site; SGD 28,000 - 69,000; Full time

Job summary

TetraMem is 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.

You will collaborate with AI researchers and hardware engineers, optimize Python or C++ code, and contribute to code reviews, testing, and documentation while pursuing a degree in CS/EE.

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 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

Enrolled in CS/EE degree

Job description

TetraMem is 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.

You will collaborate with AI researchers and hardware engineers, optimize Python or C++ code, and contribute to code reviews, testing, and documentation while pursuing a degree in CS/EE.