Summary
TetraMem is an innovative company focused on accelerating technology through novel analog compute-in-memory chips. They are seeking a Machine Learning Intern to support the development of tools and applications for neural network models, working 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
- Participate in code reviews, establish software quality standards, and contribute to automated testing, continuous integration, and technical documentation
- Troubleshoot software and system issues across the AI stack, including model conversion pipelines, runtime environments, and hardware acceleration frameworks
- Contribute to the development of innovative AI computing technologies and support research activities through technical presentations, publications, patents, and cross-functional collaboration
Skills
- 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
- Strong problem-solving skills and eagerness to learn
- Familiarity with AI/ML frameworks (e.g., TensorFlow, PyTorch) is a plus
Qualifications
Must Haves
- 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
- Strong problem-solving skills and eagerness to learn
Nice to Haves
- Familiarity with AI/ML frameworks (e.g., TensorFlow, PyTorch) is a plus