TetraMem Logo

TetraMem

US 2026 Software - Machine Learning Intern

Posted Yesterday
Be an Early Applicant
In-Office
San Jose, CA, USA
35-45 Hourly
Internship
In-Office
San Jose, CA, USA
35-45 Hourly
Internship
Develop and optimize Python or C++ code to compress, convert, deploy, and run neural network models efficiently on analog compute-in-memory chips. Profile performance, support software design and runtime environments, collaborate with AI researchers and hardware engineers, and participate in code reviews, testing, and documentation.
The summary above was generated by AI

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: $35-45 USD/hr

TetraMem celebrates diversity and is committed to creating an inclusive environment for all employees. We are proud to be an Equal Opportunity Employer and welcome applicants from all backgrounds. Qualified candidates will receive consideration for employment without regard to race, color, religion, creed, sex, gender identity or expression, sexual orientation, national origin, ancestry, age, marital status, medical condition, disability, genetic information, military or veteran status, or any other characteristic protected by applicable federal, state, or local law.

TetraMem is committed to providing reasonable accommodations to qualified applicants with disabilities throughout the recruitment process. Applicants requiring accommodation may contact Human Resources for assistance.

To ensure a fair, consistent, and efficient hiring process, all candidates must apply through TetraMem’s official ClearCompany Applicant Tracking System (ATS). Applications submitted through the ATS allow our hiring team to evaluate candidates using a standardized process and ensure timely communication throughout the recruitment process. To promote equal consideration for all applicants, applications submitted outside of the ClearCompany ATS, including direct emails, LinkedIn messages, or unsolicited submissions to employees, may not be reviewed or considered.

We encourage all interested candidates to apply through the official TetraMem Careers page.

HQ

TetraMem Newark, California, USA Office

Newark, CA, United States

Similar Jobs

Yesterday
In-Office
Sunnyvale, CA, USA
182K-242K Annually
Senior level
182K-242K Annually
Senior level
Cloud • Information Technology • Machine Learning
Debug and fix complex kernel and container runtime interactions for a large GPU fleet: investigate kernel crashes, cgroup and namespace edge cases, and scheduler/memory issues; develop and upstream kernel or container-runtime patches; build observability tooling and partner with hardware, platform, and fleet teams during incident response.
Top Skills: CCgroups V1Cgroups V2ContainerdCrashCriEbpfGdbGpu DriverKdumpKubeletKubernetesLinuxLinux KernelNamespacesRdma/RoceRunc
Yesterday
In-Office
Sunnyvale, CA, USA
153K-204K Annually
Senior level
153K-204K Annually
Senior level
Cloud • Information Technology • Machine Learning
Own and evolve a Kubernetes-native test framework that qualifies host images on real hardware. Build test harnesses, extend coverage into HPC, Slurm-on-Kubernetes and firmware/hardware, keep CI fast and reliable at fleet scale, and own results, reporting, and triage. Collaborate with kernel, firmware, imaging, and HPC teams and explore LLM-driven log triage and failure classification.
Top Skills: Argo WorkflowsCC++CiContainerizationFirmwareGoGpuInfinibandKubernetesLinuxLinux KernelLlmsPythonRoceRustSlurmSlurm-On-Kubernetes (Sunk)
Yesterday
In-Office
Sunnyvale, CA, USA
182K-242K Annually
Senior level
182K-242K Annually
Senior level
Cloud • Information Technology • Machine Learning
Design and build Linux-based execution environments for GPU workloads, extending virtualization and runtime technologies. Debug and optimize Linux kernel, hypervisor, GPU drivers, and container runtimes. Develop observability tooling, improve performance and isolation, and collaborate across security, platform, and networking teams to support multi-tenant AI infrastructure.
Top Skills: BpftraceCC++CdiContainerdCrashDmaDrgnEbpfFlame GraphsFtraceGdbGoGvisorIommuKata ContainersKdumpKernel ModulesKubevirtKvmLinuxLinux KernelMediated DevicesNested VirtualizationNumaNvidia Gpu OperatorPcie PassthroughPerfQemuRuncRustSr-IovVfioVirtio

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account