Machine Learning Intern

Numenta is looking for outstanding machine learning researchers for summer internship opportunities. The role involves participating in cutting-edge machine intelligence research and development, focusing on training and evaluating machine learning systems.

Responsibilities

• Understand the challenges of implementing performant deep learning models
• Accelerate neural network architectures through biologically inspired sparse algorithms
• Improve your software engineering skills in a collaborative environment
• Implement and optimize core learning algorithms
• Evaluate and tune current systems for both speed and accuracy

Skills

• Currently enrolled in an MS or PhD program, or a recent graduate, in Computer Science or a related field
• Outstanding programming skills in Python, especially PyTorch. Expect to work with FSDP / distributed training and be comfortable with PyTorch hooks and writing custom modules
• Deep understanding of machine learning systems, especially LLMs. For example, you are familiar with tokenizers, the transformer architecture, common training protocols, and common inference algorithms
• Capable of analyzing machine learning systems and identifying weaknesses
• Able to demonstrate strong intellectual capabilities and significant individual contributions
• Excellent written and verbal communication skills
• Experience with sparsity and sparse learning algorithms is a plus
• Experience with writing CUDA code, Triton kernels, or writing performant C++ code are pluses

Company Overview