Senior Deep Learning Framework Engineer

CommunicationsCommunicationsOtherRemoteSeniorTeam 10,001+Since 1993H1B SponsorCompany SiteLinkedIn

Location

California + 3 moreAll locations: California | North Carolina | Massachusetts | Texas

Posted

147 days ago

Salary

$152K - $241.5K / year

Seniority

Senior

Bachelor Degree5 yrs expExperience acceptedEnglishPythonPyTorch

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 Categories

Related Job Pages

More Communications Jobs

Director, Scientific Communications

Nuvalent

Nuvalent is a biotechnology research company developing precisely targeted therapies for patients with cancer. To do so, Nuvalent employs professionals across a

Communications147 days ago

• 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

Massachusetts
$220K - $250K / year
Job Closed
ASCENDING Inc. logo

Change Communications Manager

ASCENDING Inc.

AWS Certified Advanced Consulting Partner, provides Cloud Consulting/Migration/Operation, Data Analytics, IT Staffing.

Communications148 days ago
OtherRemoteTeam 11-50H1B No Sponsor

• 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.

United States
Job Closed
RESPEC logo

Lead Strategic Communications Specialist

RESPEC

Achieving the impossible. Transforming our clients' visions into reality.

Communications149 days ago
OtherRemoteTeam 201-500H1B Sponsor

• 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.

District of Columbia + 1 moreAll locations: District of Columbia | Washington
Growth Leads logo

Communications Specialist

Growth Leads

Growth Leaders in iGaming Affiliate Marketing | Fueling Passion-Driven Success in the Gaming Ecosystem

Communications149 days ago
Full TimeRemoteTeam 11-50Since 2020H1B No Sponsor

• 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.

Malta
Job Closed