Machine Learning Researcher working with a diverse team to integrate ML techniques into fusion technology. Collaborating with scientists and engineers to drive innovative energy solutions.
About the role
- Senior Engineer in generative AI focused on foundation models and AI systems design at NVIDIA. Collaborating on large-scale training and contributing to open-source projects.
Responsibilities
- Design and post-train foundation models (LLMs, VLMs, VLAs and DiTs)
- Contribute to large-scale training infrastructure and high-efficiency inference pipelines
- Collaborate on open-source and internal projects, author technical papers or patents
- Prototype and iterate rapidly on experiments across AI domains
- Design and implement model distillation algorithms for size reduction and diffusion step optimization
- Profile and benchmark training and inference pipelines
Requirements
- Minimum 8 years industry or 5+ years research/postdoc in generative AI systems
- Proficiency in PyTorch, JAX, or other deep learning frameworks
- Expertise in LLMs, coding agents, diffusion models, autoregressive models, VAE/GAN architectures
- Familiarity with all variants of attention mechanisms
- Hands-on experience with large scale training (e.g., ZeRO, DDP, FSDP)
- Experience with data processing (e.g., Ray, Spark)
- MS or PhD or equivalent in Computer Science, Machine Learning, Applied Math, Physics, or related field
- 12+ years of relevant software development experience
Benefits
- Eligible for equity and benefits
Job title
Job type
Experience level
Salary
Degree requirement
Tech skills
Location requirements
Senior AI Researcher developing Aurora, an AI system for guiding financial outcomes at Moneybox. Collaborating with ML engineers and scientists to design scalable architecture and prototype solutions.
Lead AI Research Engineer at AVEVA, focusing on prototyping and validating emerging AI technologies. Collaborating with teams to drive innovation and real - world applications in industrial software.
AI Scientist developing state - of - the - art AI solutions for drug discovery and proximity - inducing molecules. Collaborating with cross - disciplinary teams on unsolved scientific challenges in machine learning.
AI Researcher / Engineer at Constructor Knowledge Labs focusing on autonomous scientific discovery and AI systems for scientific computing and materials research.
Finance Intern supporting AI research by building and executing financial models. Collaborating with senior professionals to enhance AI's understanding of financial markets.
Machine Learning Researcher enhancing AI in K - 12 education at Kiddom. Driving significant improvements in teaching experiences and student outcomes with innovative AI applications.
Build neural networks for autonomous vehicle technology at Mobileye, focusing on deep learning model design and deployment. Collaborate with teams to ensure high - impact research solutions.
As a Research Student at LILT, you'll evaluate AI models for multilingual tasks and work with leading global labs. Opportunity for publishing and innovative research in AI.
Join Saviynt's AI Research Team to develop identity security solutions for AI agents. Collaborate to innovate and implement AI technologies for identity management.