Find your dream job at Australia's leading startups and VCs

Our exceptional communities of founders and investors are constantly seeking passionate individuals like you to join their team. Find your fit in the postings below. Just browsing? Sign up to our newsletter here, and stay up to date on the latest jobs.
companies
Jobs

Deep Learning Performance Architect Intern - 2025

Excelero Storage

Excelero Storage

IT
Shanghai, China · Beijing, China
Posted on Jun 6, 2025

We are looking for a first-class Deep Learning Performance architect to join in us to shape the performance analysis infrastructures for GPUs. We build cutting-edge analysis tools and visualization frameworks that empower engineers to optimize GPU performance for Deep Learning and HPC workloads—spanning pre-silicon architectural exploration to post-silicon validation and optimization. Your work will directly shape the tools that define how NVIDIA GPUs are analyzed, tuned, and scaled for next-gen AI systems, and impact the next-gen GPUs architectures.

What you’ll be doing:

  • Architect Performance Tooling: Develop infrastructure tools/libraries for GPU performance analysis, visualization, and automated workflows used across GPU SW/HW development life cycle

  • Unlock Architectural Insights: Analyze GPU workloads to identify bottlenecks and define new hardware profiling features that enhance perf debug and profiling capabilities.

  • AI-Powered Automation: Build AI/ML-driven tools to automate performance analysis, generate perf optimization guidance, and improve user experience of profiling infrastructure.

  • Cross-Stack Collaboration: Partner with kernel developers, system software teams, and hardware architects to co-design performance-centric solutions.

  • End-to-End Optimization: Create benchmarks to validate performance improvements across AI/HPC workloads and present actionable insights.

What we need to see:

  • BS/MS+ in relevant discipline (CS, EE, Math)

  • Proficiency in C/C++ (performance-critical coding) and Python (automation/scripting, and AI/ML frameworks)

  • Strong grasp of computer architecture (pipelines, memory hierarchies) and Operating System fundamentals

  • Understand machine learning and data analysis basics, LLM techniques such as prompt engineering, fine-tuning, vector databases

  • Experience with performance modeling, architecture simulation, profiling, and analysis.

  • Self-starter who thrives in dynamic environments and manages competing priorities effectively.

Ways to stand out from the crowd:

  • Experience with developing HW performance debugging and analysis tools

  • Familiar with System Software Stack(like CUDA Driver), CUDA kernel optimization and understand GPU architecture

  • Familiarity with GPU performance profiling tools like Nsight System, Nsight Compute

  • Practical experience or projects demonstrating LLM-based code generation, automated data analysis, or workflow assistants. Prior experience with agentic LLM frameworks like Langchain and LLamaIndex.

  • Full-Stack Versatility: Skills in JavaScript, SQL, or UI/UX design for tool interfaces.