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