Jobs at TetraMem

Software Intern

at TetraMem • Internship

Location

in-office (San Jose, United States)

Duration

3 months

Compensation

$8k/month

About this Opportunity

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 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: $4,000 - $8,000/month

Find the perfect job!

Use Job Hunt AI to find the perfect job for you.

Job Hunt AI