JAX Jobs in Los Angeles, CA

Refine Results
1 - 16 of 16 Jobs

Software Engineer, Computer Vision

SpaceX

Hawthorne, California, USA

Full-time

SpaceX was founded under the belief that a future where humanity is out exploring the stars is fundamentally more exciting than one where we are not. Today SpaceX is actively developing the technologies to make this possible, with the ultimate goal of enabling human life on Mars. SOFTWARE ENGINEER, COMPUTER VISION We are looking for good people and driven, adaptable, resilient software engineers with experience developing software to accelerate quality, speed, and efficiency of manufacturing p

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

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

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

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

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

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

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

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

(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

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

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

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

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

Lead Java Full Stack Developer

Everest Technologies

Los Angeles, California, USA

Contract

8+ years of developing multitier enterprise level applications using Java EE and Spring technologies, 5+ years of developing web and enterprise applications using all layers of Java EE technology stack. 5+ years developing applications following SOA architecture and RESTful web services using Java EE, Spring/Hibernate, JAX-WS, JAX-RS, and related technologies. 5+ years of developing front-end applications using JavaScript frameworks and libraries such as React, Angular and Vue.js. Expertise in d

Tech Lead Consultant

SigmaWay

Burbank, California, USA

Contract

Apply by Email/Direct Application at Role: Tech Lead Consultant Duration: Contract to hire Interview Process: Video Interview and on-site interview Hybrid: 2 days a week on-site (Monday and Thursdays) Job Description Must have Tech lead experience Must have expereince in Java and Spring boot Offer technical direction and hands-on leadership to the team throughout all phases of the development lifecycle.Act as a mentor and resource for resolving technical challenges, ensuring alignment with organ