Machine Learning Intern (Singapore)

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, optimize, and maintain Python and C++ software for neural network model compression, conversion, deployment, and runtime environments.
• Design and implement tools to improve machine learning model efficiency, including quantization, pruning, graph optimization, and memory reduction techniques.
• Analyze and adapt machine learning models to improve compatibility, performance, and power efficiency on Compute-in-Memory (CIM) and Analog AI hardware architectures.
• Develop software frameworks, APIs, and runtime components to support end-to-end AI model deployment on edge AI and embedded systems.
• Perform software design, implementation, debugging, testing, and performance profiling to optimize system throughput, latency, and resource utilization.
• Collaborate closely with AI researchers, hardware engineers, and system architects to validate hardware-software co-design and system-level functionality.
• Evaluate emerging machine learning models, algorithms, and deployment techniques to enhance AI inference performance on next-generation semiconductor platforms.
• Contribute to the development of innovative AI computing technologies and support research activities through technical presentations, publications, patents, and cross-functional collaboration.

Qualifications

• Currently pursuing a degree in Computer Science, Electrical Engineering, or related field.
• Solid programming experience in Python, C++, or similar.
• Familiarity with 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.

Pay

Hourly Rate: $35.00 – $45.00 per hour (USD)

Schedule

The final compensation will be determined based on the candidate’s education, experience, technical skills, and overall qualifications.

Work Location

This position is based on-site at TetraMem’s U.S. office in San Jose, California.