AI/ML Engineer

• Partake in every phase of software and model development - from ideation, design, and development, to testing and launch
• Build and improve the organization s proprietary tools
• Design, develop, and support features in Unreal Engine and Omniverse Create using C++ by writing high-quality and stable code
• Write effective code that is highly performative, stable, and can be maintained
• Work proactively to identify and solve problems and troubleshoot technical bottlenecks
• Provide support and resolve problems for the internal team
• Effectively communicate and collaborate within the engineering team to review codes/designs and fix bugs
• Proactively seek and act on feedback received from peers

Job Requirements:

• Bachelor s/Master s degree in Engineering, Computer Science (or equivalent experience)
• At least 8+ years of relevant experience as an AI/ML Engineer
• Extensive experience with programming languages, such as Python, C++, Java, or R
• Prolific experience working with distributed training and inferencing frameworks such as Horovod, Deepspeed, Lighting, Hugging Face Training, Mosaic Composer, and Megatron
• Expertise with AI/ML frameworks like TensorFlow, JAX, Pytorch, etc would be a plus
• Understanding of acceleration platforms such as TPU, GPU, APU, and FPGAs in preferred
• Previous experience working with deep learning models and associated AI frameworks, including, PyTorch, Jax (FLAN), Lightning, Deepspeed, and Collasal.ai
• Ability to skillfully write efficient AI code
• Strong interpersonal skills and great command of English communication
• Ability to communicate effectively with internal and external team members