JAX Jobs in Tennessee

Refine Results
1 - 14 of 14 Jobs

Sr. ML Engineer - AI Incubation

Oracle Corporation

•

Nashville, Tennessee, USA

Full-time

Job Description Oracle Cloud Infrastructure's Incubations team is looking for a Senior Machine Learning Engineer with strong ML experience to join a high-impact, fast-moving team building the next generation of AI-powered services at scale. In this role, you'll work across the stack-from Kubernetes infrastructure to cutting-edge ML frameworks like vLLM, LangChain, PyTorch and popular open-source models -to prototype and deliver agentic AI applications and ML infrastructure as a service. You'll
Posted 10 days ago | Updated 9 hours ago

Senior Machine Learning Engineer

Randstad - Torc

•

Remote

Contract

We are looking for a Senior Machine Learning Engineer to join our team and drive the development of advanced AI-driven solutions. You will be responsible for designing, deploying, and optimizing machine learning models that power real-world applications. This role requires deep expertise in model architecture, data engineering, MLOps, and production-scale AI systems. What you'll do:Design, develop, and deploy scalable, production-ready machine learning models for real-world applications.Architec
Posted 60+ days ago | Updated 8 days ago

Principal Software Engineer, TensorRT-LLM

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

We are now looking for a Principal Software Engineer, TensorRT-LLM ! NVIDIA is hiring experienced principal software engineer for its TensorRT-LLM team. Academic and commercial groups around the world are using GPUs to power a revolution in AI, enabling breakthroughs in areas like content creation, code generation and information synthesis, that have put AI on the precipice of an "iPhone moment". Join the team building the AI serving software which is foundational to product lines within NVIDIA
Posted 9 days ago | Updated 9 hours ago

Developer Technologies Engineer, Robotics Reinforcement Learning

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

Today, NVIDIA is tapping into the unlimited potential of AI to define the next era of computing. An era in which our GPU acts as the brains of computers, robots, and self-driving cars that can understand the world. Doing what's never been done before takes vision, innovation, and the world's best talent. As an NVIDIAN, you'll be immersed in a diverse, encouraging environment where everyone is inspired to do their best work. Come join the team and see how we can make a lasting impact on the world
Posted 23 days ago | Updated 9 hours ago

Senior GenAI Algorithms Engineer

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

We are now looking for a Senior Gen AI Algorithms Engineer! NVIDIA is seeking engineers to design, develop and optimize Artificial Intelligence solutions to diverse real-world problems. If you have a strong understanding of deep learning and in particular large language models and their multimodal variants, then this role may be a great fit for you! Collaborate and interact with internal partners, users, and members of the open source community to analyze, define and implement highly optimized A
Posted 36 days ago | Updated 9 hours ago

Principal Deep Learning Software Engineer, LLM Performance

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

We are now looking for a Principal Deep Learning Software Engineer, LLM Performance! NVIDIA is seeking an experienced Deep Learning Engineer passionate about analyzing and improving the performance of LLM inference! NVIDIA is rapidly growing our research and development for Deep Learning Inference and is seeking excellent Software Engineers at all levels of expertise to join our team. Companies around the world are using NVIDIA GPUs to power a revolution in deep learning, enabling breakthroughs
Posted 44 days ago | Updated 9 hours ago

Senior Deep Learning Software Engineer, Inference and Model Optimization

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

NVIDIA is at the forefront of the generative AI revolution! The Algorithmic Model Optimization Team specifically focuses on optimizing generative AI models such as large language models (LLM) and diffusion models for maximal inference efficiency using techniques ranging from neural architecture search and pruning to sparsity, quantization, and automated deployment strategies. Our work includes conducting applied research to improve model efficiency as well as developing an innovative software pl
Posted 36 days ago | Updated 9 hours ago

Senior Deep Learning Software Engineer, LLM Performance

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

We are now looking for a Senior Deep Learning Software Engineer, LLM Performance! NVIDIA is seeking an experienced Deep Learning Engineer passionate about analyzing and improving the performance of LLM inference! NVIDIA is rapidly growing our research and development for Deep Learning Inference and is seeking excellent Software Engineers at all levels of expertise to join our team. Companies around the world are using NVIDIA GPUs to power a revolution in deep learning, enabling breakthroughs in
Posted 44 days ago | Updated 9 hours ago

Senior DGX Cloud AI Infrastructure Software Engineer

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

Joining NVIDIA's DGX Cloud AI Efficiency Team means contributing to the infrastructure that powers our innovative AI research. This team focuses on optimizing efficiency and resiliency of AI workloads, as well as developing scalable AI and Data infrastructure tools and services. Our objective is to deliver a stable, scalable environment for AI researchers, providing them with the necessary resources and scale to foster innovation. We are seeking an AI infrastructure software engineer to join our
Posted 46 days ago | Updated 9 hours ago

(USA) Senior, Data Scientist

Walmart Inc.

•

Remote or New York, New York, USA

Full-time

Position Summary Join Walmart|VIZIO and take your career to the next level! We are looking for a highly talented and motivated Data Scientist to help deliver strategic initiatives across the organization. We look for people who can effectively partner with Product Management, Engineering, and business leaders to deliver easy, effective, and insightful analytics. We are a growing team, focusing on challenging problems that make a difference to our business. We concentrate on high-impact, high-va
Posted 49 days ago | Updated 9 hours ago

Machine Learning Software Engineer L4/L5

Netflix, Inc.

•

Remote

Full-time

Netflix is one of the world's leading entertainment services, with 283 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. As Netflix continues to grow, so do the opportunities to enhance our personalization systems and algorithms. We're looking for a passionate and talented Software Engineer fo
Posted 60+ days ago | Updated 9 hours ago

Senior Math Libraries Engineers - Python APIs

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

NVIDIA is looking for a self-motivated and specialist software engineer for the design and development of high-performant Python APIs for our math libraries. In the last decade, Python has become the de-facto leading programming language for engineers in AI and data science, and more recently in HPC and scientific computing. NVIDIA has been at the forefront of providing GPU accelerated Deep Learning frameworks. These frameworks provide an efficient high-level programming interface allowing their
Posted 60+ days ago | Updated 9 hours ago

Software Engineer Staff

Lockheed Martin Corporation

•

Remote or Colorado Springs, Colorado, USA

Full-time

Job Description Space is a critical domain, connecting our technologies, our security and our humanity. While others view space as a destination, we see it as a realm of possibilities, where we can do more - we can innovate, invest, inspire and integrate our capabilities to transform the future. At Lockheed Martin Space, we aim to harness the full potential of space to cultivate innovation, reduce costs, and push the boundaries of what technology can achieve. We're creating future-ready solutio
Posted 38 days ago | Updated 9 hours ago

Software Engineering Manager, Sparse Linear Algebra Libraries

NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

We are looking for a software engineering manager to lead our development efforts for high-performance sparse linear algebra libraries in the AI and scientific computing domains. Around the world, leading commercial and academic organizations are revolutionizing AI, data analytics, and scientific and engineering simulations, using data centers powered by GPUs. Applications of these technologies include LLMs, computer vision, encryption, computer aided engineering (CAE), electronic design automat