Senior Deep Learning Algorithm Engineer

This role is based in Santa Clara, CA with remote options available.

Compensation:

$148,000 - $287,500/year

Responsibilities:

• Develop algorithms for AI/DL, data analytics, machine learning, or scientific computing.
• Contribute to Megatron Core and NeMo Framework.
• Address large-scale AI training and inference challenges.
• Innovate model architectures and distributed training algorithms.
• Performance tuning and optimizations on NVIDIA GPU architectures.

Requirements:

• MS or equivalent experience in Computer Science, AI, Applied Math, or related fields with 3+ years of experience.
• Experience with AI Frameworks (e.g. PyTorch, JAX).
• Proficient in Python programming and software design.
• Strong understanding of AI/Deep-Learning fundamentals.

Ways to stand out:

• Experience in large-scale AI training and performance analysis.
• Expertise in distributed computing and mixed precision training.
• Contributions to open source deep learning frameworks.