ML Software Intern — AI Hardware & Model Optimization

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

Job summary

TetraMem seeks a Software Intern to help develop tools, frameworks, and applications enabling neural network models to run efficiently on its novel analog compute-in-memory chips. Join a cross-disciplinary team at the intersection of software, hardware, and AI in San Jose.

You will work on Python/C++ code for model compression, deployment, and runtime environments, adapt ML models for compute-in-memory hardware, and collaborate with AI researchers and engineers to validate system functionality.

Qualifications

• Pursuing degree in CS/EE or related field.
• Proficient in Python or C++.
• Familiarity with ML frameworks like TensorFlow or PyTorch.

Responsibilities

• Develop and optimize Python or C++ code for NN and ML model compression, conversion, deployment, and runtime environments.
• Analyze and adapt ML models to improve compatibility and efficiency on compute-in-memory hardware.
• 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

AI/ML frameworks

Education

BS in Computer Science or related field

Tools

TensorFlow

PyTorch

Job description

TetraMem seeks a Software Intern to help develop tools, frameworks, and applications enabling neural network models to run efficiently on its novel analog compute-in-memory chips. Join a cross-disciplinary team at the intersection of software, hardware, and AI in San Jose.

You will work on Python/C++ code for model compression, deployment, and runtime environments, adapt ML models for compute-in-memory hardware, and collaborate with AI researchers and engineers to validate system functionality.