Flag job

Report

Software Intern

Salary

$48k - $96k

Min Experience

0 years

Location

San Jose, CA

JobType

internship

About the job

Info This job is sourced from a job board

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

Skills

python
c++
tensorflow
pytorch