Senior GenAI Framework Engineer
NVIDIA
CloudOpen SourcePythonPyTorch
Senior Applications Engineer, GenAI for Science
NVIDIA
full-time
Posted on:
Origin: • 🇺🇸 United States • California
Visit company websiteSalary
💰 $124,000 - $230,000 per year
Job Level
Senior
Tech Stack
PythonPyTorch
About the role
- Work alongside research teams worldwide adopting NVIDIA's GenAI software for scientific data/models.
- Debug, profile and recommend performance optimizations for GenAI models and workflows.
- Provide customer feedback to engineering teams to improve NVIDIA’s GenAI software for science research.
- Participate in GenAI hands-on hackathons and workshops to drive adoption of NVIDIA’s GenAI stack in the scientific community.
- Collaborate with leading science and research institutes (including in Japan) to promote GPU-accelerated computing solutions and deploy GenAI models at scale.
- Help collaborators convert proofs-of-concept into scalable solutions for scientific discovery.
Requirements
- Excellent verbal, written communication, and technical presentation skills in Japanese.
- Business level English communication.
- Degree or equivalent experience in Computer Science, Applied Mathematics, or related engineering field (Ph.D. or Masters preferred).
- 5+ years hands-on experience developing, optimizing and running AI/ML models or LLMs using popular frameworks (PyTorch or similar) and languages (C++, Python).
- Deep understanding of Gen AI model architectures (GPT-x, Llama, MoE, etc).
- Ability to work independently and as part of a globally distributed team.
- Expertise in deploying large-scale training and inferencing pipelines.
- Hands-on experience customizing AI models for multi-modal scientific data.
- Contributions to cornerstone research papers or open-source projects involving GenAI/AI for science applications.
- Experience integrating simulation, AI/GenAI models and running them on large-scale HPC systems.
