Intern - Machine Learning Engineering

Description

The company makes memory and storage products used in AI, high-performance computing and data-centric systems.

You would join the pathfinding group, which studies future memory and storage technologies, emerging AI workloads and system designs to help shape the company's long-term roadmap.

This is an internship.

Responsibilities

• Study upcoming AI models and compute designs to work out what future memory and storage will need.
• Weigh performance, power, cost, bandwidth, capacity, latency and scalability against each other across AI systems.
• Build models, simulations, benchmarks and analysis tools to test new memory and storage ideas.
• Measure how AI workloads behave on CPUs, GPUs, accelerators and memory-centric platforms.
• Work with architects and engineers to judge new technologies, report results and support proof-of-concept tests.

Requirements

• Currently enrolled in a Bachelor's, Master's or PhD program in computer engineering, electrical engineering, computer science or a closely related technical field.
• 0-2 years of relevant experience from internships, research, coursework or technical projects in computer architecture, AI/ML, high-performance computing or memory systems.
• A solid grasp of computer architecture, machine learning, AI or high-performance computing concepts.
• Working knowledge of Python and modern AI/ML frameworks such as PyTorch, TensorFlow, JAX or similar.
• Strong analytical, problem-solving, communication and teamwork skills.
• Preferred: understanding of memory systems, caching, memory hierarchies, DRAM and performance bottlenecks.
• Preferred: experience with performance analysis, benchmarking, profiling, simulation or system modeling.
• Preferred: familiarity with GPU computing, accelerators, heterogeneous systems or AI infrastructure.
• Preferred: understanding of AI model architectures, training, inference and workload characterization.
• Preferred: experience using AI tools for research, software development, data analysis or engineering productivity.

Conditions

Details on your right to work are linked from the posting.

The company does not charge candidates recruitment fees, prohibits child labor and follows applicable labor laws.

Benefits

• Health cover spanning dental, vision and medical, chosen to fit your family and budget.
• Coverage that protects part of your income if illness or injury keeps you from working.
• Paid family leave.
• Paid time off and paid holidays.
• The benefits guide is linked from the posting.