JAX jobs in Pennsylvania

Google Cloud Platform Infra SME with hands on HPC Experience

Remote

•

Today

Role: Google Cloud Platform Infra SME with hands on HPC Experience Location : US Remote Experience: 10 years What is Needed Google Cloud Professional Architect Certification is MandatoryGoogle Cloud Platform SME with hands-on HPC experience (with Infra)GPU / TPU ExperienceFamiliarity with hands on IaC is a mustTerraformGKENetworkingStoragePythonLibrary familiarity with: numpy/pandas/Pytorch/JAX, including optimizationExperience with Nvidia and/or Google TPU hardware in GCE and GKEML specific Goo

Easy Apply

Contract

Depends on Experience

Senior Software Engineer, Compute Infrastructure for Robotics Research

Remote or Santa Clara, California

•

Today

We are seeking a Senior Software Engineer to join a new team building the foundational infrastructure for Robotics Research. This new team will work very closely with NVIDIA's Generalist Embodied Agent Research (GEAR) group. The near term focus is Project GR00T, NVIDIA's moonshot initiative at building foundation models and full-stack technology for humanoid robots. This particular position will focus on compute infrastructure. You will work with an amazing and collaborative research team that

Full-time

USD 148,000.00 per year

Senior Research Engineer

Remote or Santa Clara, California

•

Today

Join NVIDIA and help build the software that will define the future of generative AI. We are looking for a research engineer who is passionate about open-source and excited to create our next-generation post-training software stack. You will work at the intersection of research and engineering, collaborating with the Post-Training and Frameworks teams to invent, implement, and scale the core technologies behind our Nemotron models. What you'll be doing: Work with applied researchers to design,

Full-time

USD 160,000.00 - 258,750.00 per year

Senior Math Libraries Engineer, CPU and GPU Optimization

Remote or Santa Clara, California

•

Today

NVIDIA is looking for an expert software engineer to help us deliver CUDA-X libraries across the NVIDIA CPU and GPU ecosystem. For over a decade, NVIDIA's accelerated computing platform has revolutionized HPC and AI with applications ranging from COVID-19 research to autonomous machines. Did you know that our team develops the GPU/CPU-accelerated mathematical libraries that make all of this possible? The hardware and software accelerated computing ecosystem is constantly evolving, including shi

Full-time

USD 224,000.00 per year

Senior Deep Learning Software Engineer - Autonomous Vehicles

Remote or Santa Clara, California

•

Today

We are looking for outstanding Deep Learning Software Engineers to develop and productize NVIDIA's deep learning solutions in autonomous driving vehicles. As a member of our Solution Engineering-Automotive Machine Learning team, you will apply ground breaking NVIDIA deep learning model training/inference software libraries for deployment on NVIDIA's hardware architecture. You will develop new deep learning architectures, train deep learning models, and compile and optimize DNN graphs. As a part

Full-time

USD 148,000.00 per year

Senior Software Engineer, MathDx - cuBLASDx and cuSolverDx

Remote or Santa Clara, California

•

Today

NVIDIA is looking for an expert software engineer to help us expand our catalog of Device eXtension (Dx) APIs for our math libraries. For over a decade, NVIDIA's accelerated computing platform has revolutionized HPC and AI with applications ranging from COVID-19 research to autonomous machines. Did you know that our team develops the GPU-accelerated mathematical libraries that make all of this possible? The hardware and software accelerated computing ecosystem is constantly evolving, including

Full-time

USD 148,000.00 per year

Senior Software Engineer, Data Infrastructure for Robotics Research

Remote or Santa Clara, California

•

Today

We are seeking a Senior Software Engineer to join a new team building the foundational infrastructure for Robotics Research. This new team will work very closely with NVIDIA's Generalist Embodied Agent Research (GEAR) group. The near term focus is Project GR00T, NVIDIA's moonshot initiative at building foundation models and full-stack technology for humanoid robots. This particular position will focus on data infrastructure. You will work with an amazing and collaborative research team that con

Full-time

USD 224,000.00 per year

Machine Learning Engineer 5 - Content & Studio

Remote

•

Today

Netflix is one of the world's leading entertainment services, with over 300 million paid memberships in over 190 countries enjoying TV series, films and games across a wide variety of genres and languages. Members can play, pause and resume watching as much as they want, anytime, anywhere, and can change their plans at any time. At Netflix, we seek to entertain the world. We launch thousands of new TV shows and movies every year for our members across the globe. To do so, we need to understand

Full-time

USD 150,000.00 per year

Senior Software Engineer, Machine Learning Inference

Remote or Santa Clara, California

•

Today

At NVIDIA, we're at the forefront of innovation, driving advancements in AI and machine learning to solve some of the world's most challenging problems. We're seeking talented and motivated engineers to join our TensorRT team in developing the industry-leading deep learning inference software for NVIDIA AI accelerators. As a Senior Software Engineer in the TensorRT team, you will be responsible for designing and implementing inference software optimizations to power AI applications on NVIDIA GPU

Full-time

USD 184,000.00 per year

Senior HPC Performance Engineer - AI for Science at Scale

Remote or Santa Clara, California

•

Today

NVIDIA has become the platform upon which every new AI-powered application is built. We are seeking a Sr. HPC Performance engineer to join our team of scientists and engineers passionate about building the next generation of scientific machine learning (ML) frameworks. Starting with digital biology, through high performance computing (HPC) and powerful ML methods, together, we will advance NVIDIA's capacity to accelerate AI for Science and industries that depend on it. What you'll be doing: De

Full-time

USD 184,000.00 per year

Senior MLOps Engineer

Remote or Santa Clara, California

•

Today

NVIDIA is seeking a Senior MLOps Engineer to help design and scale the infrastructure that powers our AI research and product development. In this role, you will partner closely with research scientists and product teams to accelerate their success on the latest GPU and accelerator platforms. By building robust ML pipelines and scalable systems, you will ensure that cutting-edge hardware innovations translate directly into faster experiments, more efficient training, and reproducible deployments

Full-time

USD 184,000.00 per year

Systems Software Engineer, AI Infrastructure

Remote or Santa Clara, California

•

Today

NVIDIA is widely considered to be one of the technology world's most desirable employers. We have some of the most forward-thinking and hardworking people in the world working for us. If you're creative and autonomous, we want to hear from you! NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for over 30 years. It's a unique legacy of innovation that's fueled by phenomenal technology and outstanding people. Today, we're tapping into the unlimited potential of

Full-time

USD 184,000.00 - 287,500.00 per year

Principal Artificial Intelligence Algorithms Engineer

Remote or Santa Clara, California

•

Today

NVIDIA is looking for engineers for our core AI Frameworks (Megatron Core and NeMo Framework) team to design, develop and optimize diverse real world workloads. Megatron Core and NeMo Framework are open-source, scalable and cloud-native frameworks built for researchers and developers working on Large Language Models (LLM) and Multimodal (MM) foundation model pretraining and post-training. Our GenAI Frameworks provide end-to-end model training, including pretraining, reasoning, alignment, customi

Full-time

USD 272,000.00 per year

Senior Deep Learning Algorithm Engineer, Training Framework

Remote or Santa Clara, California

•

Today

NVIDIA is looking for engineers for our NeMo Framework team to design, develop and optimize diverse real world workloads. NeMo Framework is an open-source, scalable and cloud-native framework built for researchers and developers working on Large Language Models (LLM) and Multimodal (MM) foundation model pretraining and post-training. Our GenAI Frameworks provide end-to-end model training, including pretraining, reasoning, alignment, customization, evaluation, deployment and tooling to optimize p

Full-time

USD 184,000.00 per year

Lead Data Scientist - Autonomous Goal Management

Remote

•

Today

Become a part of our caring community and help us put health first The Enterprise AI organization at Humana is a pioneering force, driving AI innovation across our Insurance and CenterWell business segments. By collaborating with world-leading experts, we are at the forefront of delivering cutting-edge AI technologies for improving care quality and experience of millions of consumers. We are actively seeking top talent to develop robust and reusable AI modules and pipelines, ensuring adherence t

Full-time

USD 142,300.00 - 195,700.00 per year

Lead Data Scientist - Safety Alignment

Remote

•

Today

Full-time

USD 142,300.00 - 195,700.00 per year

Java Full Stack Developer

Remote or Tampa, Florida

•

Today

Join Citi, a leading global bank, in shaping the future of financial technology. Our Global Technology team delivers innovative solutions to nearly 200 million customer accounts worldwide. We value talent, passion, and a commitment to excellence in a dynamic and collaborative environment. The Opportunity: This pivotal role blends deep technical leadership with strong analytical and business understanding. You will be responsible for leading the technical direction of critical applications, whi

Full-time

USD 113,840.00 - 170,760.00 per year

Lead Data Scientist - AI-Human Interaction

Remote

•

Today

Full-time

USD 142,300.00 - 195,700.00 per year

Senior Cloud Services Software Engineer

Remote or Santa Clara, California

•

Today

Joining NVIDIA's DGX Cloud Team means contributing to the infrastructure that powers our innovative AI research. This team focuses on optimizing efficiency and resiliency of ML workloads, as well as developing scalable AI infrastructure tools and services. Our objective is to deliver a stable, scalable environment for AI researchers, providing them with the vital resources and scale to champion innovation. We are seeking a distributed software engineer to join our team! As a Senior engineer, yo

Full-time

USD 224,000.00 per year

Staff Software Engineer, Machine Learning

Remote or San Francisco, California

•

Today

Discord is used by over 200 million people every month for many different reasons, but there's one thing that nearly everyone does on our platform: play video games. Over 90% of our users play games, spending a combined 1.5 billion hours playing thousands of unique titles on Discord each month. Discord plays a uniquely important role in the future of gaming. We are focused on making it easier and more fun for people to talk and hang out before, during, and after playing games. We're currently l

Full-time

USD 272,000.00 - 306,000.00 per year

Filter Results

Job post features

Posted date

Work settings

Employment type

Employer type

Work authorization

JAX jobs in Pennsylvania, USA