AI/ML Systems Research Intern

Nokia Bell Labs is a global leader in connectivity for the AI era, advancing connectivity to secure a brighter world. They are seeking an AI/ML Systems Research Intern to design and implement state-of-the-art AI/ML decentralized systems while validating and evaluating implementations in their labs.

Responsibilities

• Design and implement state-of-the-art AI/ML decentralized systems
• Validate and evaluate your implementation in our cutting-edge labs
• Interface, explore, and learn from the experts

Skills

• Currently a candidate for a PhD in Computer Science, Computer Systems Engineering, Math, Artificial Intelligence, or related field at an accredited school in the USA
• Expertise in deep learning fundamentals, including large language models and agent-based systems, and experience with training, deploying, and/or profiling models
• Experience in principled systems design and development
• Excellent communication skills, with the ability to analyze complex problems and effectively communicate findings
• Strong publication record in top-tier AI and systems conferences
• Familiarity with large-scale model inference and optimization, as well as experience in LLM reasoning, prompt engineering, and resource-constrained computation
• Experience managing GPU or accelerator resources, optimizing performance, and benchmarking across different hardware environments
• A solid understanding of AI infrastructure design and inference workflows—such as KV-cache management, batching, and offloading
• Knowledge of computational graph representations (e.g., ONNX, MLIR, XLA, TorchScript) and model optimization frameworks (e.g., TensorRT, TVM)
• Experience working with heterogeneous accelerator ecosystems (e.g., TPUs, AMD ROCm GPUs) or parallelizing compilers
• Understanding of distributed or edge inference systems (e.g., Ray Serve, DeepSpeed-Inference, vLLM), with familiarity in blockchain technologies, smart contracts, or wireless networking protocols (Wi-Fi, 3GPP, Bluetooth)

Benefits

• Flexible and hybrid working schemes to balance study, work, and life
• Professional development events and networking opportunities
• Well-being programs, including Personal Support Service 24/7 - a confidential support channel open to all Nokia employees and their families in challenging situations
• Opportunities to join Nokia Employee Resource Groups (NERGs) and build connections across the organization
• Employee Growth Solutions, mentorship programs, and coaching support for your career development
• A learning environment that fosters both personal growth and professional development – for your role and beyond

Company Overview