Senior Deep Learning Framework Engineer
Location
California + 3 moreAll locations: California | North Carolina | Massachusetts | Texas
Posted
147 days ago
Salary
$152K - $241.5K / year
Seniority
Senior
Job Description
Senior Deep Learning Framework Engineer
NVIDIA
• Integrate new communication libraries features in AI frameworks: from PoC to performance analysis to production • Perform deep analysis of AI workloads and frameworks to identify multi-GPU communication requirements and opportunities. • Collaborate hands-on with teams working on the latest AI models. • Improve AI compilers to hide communications or perform automatic fusion. • Conduct in-depth AI workload performance characterization on multi-GPU clusters. • Design fault-tolerant and elastic solutions for large-scale or dynamic AI workloads. • Author custom communication or fused compute-communication kernels to showcase ultimate performance on NV platforms. • Influence the roadmap of communication libraries - NCCL & NVSHMEM. • Collaborate with a very dynamic team across multiple time zones.
Job Requirements
- B.S, M.S. or PHD in Computer Science, or related field (or equivalent experience) with 5+ software engineering and HPC/AI experience
- Development or integration experience with Deep Learning Frameworks such PyTorch, JAX, and Inference Engines such as TRT-LLM, vLLM, SGLang
- Rapid prototyping and development with Python, C++, CUDA or related DSLs (Triton, cuTe)
- Solid grasp of AI models, parallelisms, and/or compiler technologies (e.g. torch.compile)
- Experience conducting performance benchmarking on AI clusters.
- Familiarity with at least one performance profiler toolchain (PyTorch profiler, NVIDIA Nsight Systems)
- Understanding of HPC/AI communication concepts (1-sided v 2-sided communication, elasticity, resiliency, topology discovery, etc)
- Adaptability and passion to learn new areas and tools
- Flexibility to work and communicate effectively across different teams and timezones
Benefits
- equity
- benefits
Related Guides
Related Categories
Related Job Pages
More Communications Jobs
Director, Scientific Communications
NuvalentNuvalent is a biotechnology research company developing precisely targeted therapies for patients with cancer. To do so, Nuvalent employs professionals across a
• Lead asset publication and communications strategy, planning, and execution • Collaborate cross-functionally on publication planning and execution • Partner with Clinical Development, Medical Affairs, Evidence Generation, and Brand teams to identify and prioritize data gaps and publication concepts • Co-lead Publication Planning Committee, including driving agendas, documenting meeting minutes, and managing action items • Manage approval of publication strategy and plans by the Publication Planning Committee and endorsement by PDT and LT • Oversee all aspects of publication development for abstracts, congress presentations, manuscripts, reviews, publication extenders, etc. • Critically review drafts and adjudicate comments • Drive development and updates of the asset scientific platform and lexicon in collaboration with cross-functional teams; ensure utilization to drive consistency across all related medical materials and channels • Support portfolio-wide publications as needed • Publication management platform oversight (eg, Datavision or similar platform) • Support data dissemination strategy and plans across various Medical Affairs initiatives, including content development, congress activities, and medical education initiatives. • Subject matter expert on disease state, competitive landscape, and publication benchmarks • Manage vendors, ensuring adherence to timelines, process, and publication best practices
Change Communications Manager
ASCENDING Inc.AWS Certified Advanced Consulting Partner, provides Cloud Consulting/Migration/Operation, Data Analytics, IT Staffing.
• Develop and execute communications to boost engagement and alignment. • Craft communication plans aligned with goals. • Produce materials that clarify and amplify key messages. • Support internal campaigns and collaborate with internal partners to manage feedback and deliver integrated outputs.
Lead Strategic Communications Specialist
RESPECAchieving the impossible. Transforming our clients' visions into reality.
• Develop and maintain an executive-level communications plan. • Define key messages, themes, channels, and target audiences for OCI initiatives. • Oversee the design and delivery of digital, print, and multimedia products. • Ensure all communications reflect the organization’s values, tone, and visual identity. • Develop strategies for stakeholder engagement, including town halls, podcasts, and office hours. • Translate complex technical topics into accessible language. • Manage a centralized knowledge repository to support stakeholder communication and clinical requirement definition.
Communications Specialist
Growth LeadsGrowth Leaders in iGaming Affiliate Marketing | Fueling Passion-Driven Success in the Gaming Ecosystem
• Develop and distribute communication materials including press releases, articles, organizational announcements, website content, and success stories. • Ensure consistent messaging and positioning across all external platforms and communication channels. • Support media engagement and external partnerships to strengthen visibility and public profile. • Contribute to maintaining a positive and professional public image through timely and accurate communications. • Support management of the organization’s digital footprint, including website updates and coordination of social and digital content aligned with organizational priorities. • Collaborate with marketing and growth teams to align digital campaigns and communication activities with business objectives. • Monitor engagement and support reporting on communication performance where applicable. • Support communication planning and execution for events, conferences, partner engagements, and public-facing activities. • Coordinate event-related content, announcements, invitations, and post-event communications. • Work with internal stakeholders to ensure events support outreach, partnerships, and growth objectives. • Prepare and distribute internal communications such as newsletters, staff updates, leadership messages, and initiative announcements. • Support teams in communicating key projects, milestones, and organizational priorities to staff. • Help ensure employees are informed about external activities and initiatives that affect the organization.



