英伟达Senior CUDA Test Development Software Engineer

NVIDIA is now looking for LLM Train Framework Engineers for the Megatron Core team. Megatron Core is open-source, scalable, and cloud-native frameworks built for researchers and developers working on Large Language Models (LLM) and Multimodal (MM) foundation model pretraining and post-training. Our GenAI Frameworks provide end-to-end model training, including pretraining, alignment, customization, evaluation, deployment, and tooling to optimize performance and user experience. Build on Megatron Core Framework's capabilities by inventing advanced distributed training algorithms and model optimizations. Collaborate with partners to implement optimized solutions. What you’ll be doing: • Build and develop open source Megatron Core. • Address extensive AI training and inference obstacles, covering the entire model lifecycle including orchestration, data pre-processing, conducting model training and tuning, and deploying models. • Work at the intersection of AI applications, libraries, frameworks, and the entire software stack. • Spearhead advancements in model architectures, distributed training strategies, and model parallel approaches. • Enhance the pace of foundation model training and optimization through mixed precision formulas and advanced NVIDIA GPU structures. • Performance tuning and optimizations of deep learning framework and software components. • Research, prototype, and develop robust and scalable AI tools and pipelines.

• Design and implement functional/performance tests for CUDA products, like driver and library. • Automate CUDA tests, design test plans and integrate into automation testing infrastructure. • Triage test results, root cause test failures or performance drops, and drive through bugs to fix. • Develop scripts/tools and optimize workflow to improve efficiency and productivity.

• Design and implement functional/performance tests for CUDA products, like driver and library.  • Automate CUDA tests, design test plans and integrate into automation testing infrastructure.  • Triage test results, root cause test failures or performance drops, and drive through bugs to fix.  • Develop scripts/tools and optimize workflow to improve efficiency and productivity.

We are looking for a Software Test development engineer in NVIDIA’s AI SWQA team.The position is in NVIDIA AI Software Quality Assurance team that defines, develops and performs tests to validate robustness and measure the performance of NVIDIA‘s AI software and GPU Infrastructure for autonomous driving, healthcare, speech recognition, natural language processing, and a wide variety of other AI scenarios. This team collaborates with multiple AI product teams to develop new products; derive and improve complex test plans; and improve our workflow processes for a diverse range of GPU computing platforms. You should grow with being in the critical path supporting developers working for billion-dollar business lines as well as intimately understanding the values of responsiveness, thoroughness and teamwork. You should constantly foster and implement efficiency improvements across your domain. Join the team which is building software which will be used by the entire world! What you’ll be doing: • Work closely with global cross-functional teams to understand the test requirements and take ownership of product quality. • Plan/design/execute/report/automate test plan/test case/test reports. • Manage bug lifecycle and co-work with inter-groups to drive for solutions. • Automate test cases and assist in the architecture, crafting and implementing of test frameworks. • In-house repro and verify customer issues/fixes.