JAX Jobs

Refine Results
101 - 120 of 188 Jobs

Sr. Software Engineer TS/SCI Polygraph

Leidos

•

Reston, Virginia, USA

Full-time

Description Leidos has an exciting opportunity for a Sr. Software Engineer! *Must have an active TS/SCI Polygraph up front. No exceptions.* You will perform software development lifecycle (SDLC) activities as both an individual and a member of our top-notch agile development team building a large complex enterprise system. Development includes the full range of turning Agile user stories into implementable concepts, through development, unit testing, integration and test, and deployment of the
Posted 10 days ago | Updated 3 hours ago

Sr. Applied ML Engineer, Apple Services Localization Engineering

Apple, Inc.

•

No location provided

Full-time

Are you passionate about building extraordinary products and embracing state of art techniques and technologies? In Apple Services Localization Engineering, we are leveraging Machine Learning and Machine Translation to expand our global product reach and bring our Services to more users around the world. The Apple Services Engineering team is one of the most exciting examples of Apple's passion for combining art and technology. These are the people who power Apple Music, App Store, Apple TV, App
Posted 57 days ago | Updated 4 hours ago

Senior High-Performance LLM Training Engineer

NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

We are now looking for a Senior High-Performance LLM Training Engineer! NVIDIA is seeking experienced engineers specializing in performance analysis and optimization to improve the efficiency of LLM training workloads, which are shaping the world's most advanced computing systems. This position focuses on optimizing NVIDIA's high-performance LLM software stack in frameworks like PyTorch and JAX for high-performance training on thousands of GPUs, while also helping shape hardware roadmaps for th
Posted 60+ days ago | Updated 4 hours ago

Software Engineer III - Machine Learning Engineer

JPMorgan Chase & Co.

•

Wilmington, Delaware, USA

Full-time

Job Description Are you ready to make a significant impact in the world of AI and machine learning? At JPMorgan Chase, In this role, you'll be a key player in an agile team dedicated to enhancing, building, and delivering market-leading technology products that are secure, stable, and scalable As a Software Engineer /Machine Learning Engineer, at JPMorgan Chase within the Corporate Sector - AIML Data Platforms Team, you are an integral part of an agile team that works to enhance, build, and de
Posted 42 days ago | Updated 4 hours ago

Software Engineer - Model Serving Infrastructure - USDS

TikTok

•

Seattle, Washington, USA

Full-time

Location : Seattle Employment Type : Regular Job Code : A32426A Apply to this job Share this listing: Responsibilities Team Intro The mission of our AML team is to push the next-generation AI infrastructure and recommendation platform for the ads ranking, search ranking, live & ecom ranking in our company. We also drive substantial impact on core businesses of the company. Currently, we are looking for Machine Learning Engineer - Model Serving Infrastructure to join our team to support an
Posted 60+ days ago | Updated 4 hours ago

Staff Software Engineer, ML Inference Compiler & Deployment, Optimus

Tesla Motors

•

Palo Alto, California, USA

Full-time

As a Software Engineer within our Autonomy teams, you will contribute to one of the most advanced and widely deployed AI Platforms in the world for Autopilot and our Humanoid Robot, Optimus. In this role, you will be responsible for the internal working of the AI inference stack running neural networks in Optimus and millions of Tesla vehicles. You will collaborate closely with the Optimus AI Engineers and AI Hardware Engineers to understand the full inference stack, co-design models to fit the
Posted 60+ days ago | Updated 4 hours ago

Sr Machine Learning Engineer, ML Systems Engineering

Apple, Inc.

•

No location provided

Full-time

Apple's products combine the best hardware and incredible software to deliver magical experiences to our customers. The Proactive Intelligence team builds features that anticipate customer's needs and create personalized experiences by adapting to user behaviors with machine learning running locally on users' devices. We are seeking exceptional machine learning engineers with a strong software engineering background to join our team where we're making Apple Intelligence more capable and personal
Posted 60+ days ago | Updated 4 hours ago

Director Software Development - AI Inference

AMD (Advanced Micro Devices)

•

Bellevue, Washington, USA

Full-time

WHAT YOU DO AT AMD CHANGES EVERYTHING We care deeply about transforming lives with AMD technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences - the building blocks for the data center, artificial intelligence, PCs, gaming and embedded. Underpinning our mission is the AMD culture. We push the limits of innovation to solve the world's most important challenges. We strive for execution excellen
Posted 60+ days ago | Updated 4 hours ago

AIML - Sr. Machine Learning Engineer, Siri & Information Intelligence

Apple, Inc.

•

No location provided

Full-time

Do you want to make Siri and Apple products smarter for our users? Our Siri and Information Intelligence team is redefining how hundreds of millions of people use their devices to get information. We are an Applied ML team pushing the limits on realtime augmented information retrieval and generation, information safety and search technologies, while also responsible for a few user facing production services. We are part of a wider effort to power information across a variety of Apple products -
Posted 60+ days ago | Updated 4 hours ago

AI and ML Infra Software Engineer, GPU Clusters

NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for more than 25 years. It's a unique legacy of innovation that's fueled by great technology-and amazing people. Today, we're 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.
Posted 44 days ago | Updated 4 hours ago

ML Framework (MetalLM) Engineer

Apple, Inc.

•

No location provided

Full-time

Apple's ML Frameworks (MetalLM) team in GPU, Graphics and Machine Learning works on enabling Apple Intelligence through high-performance, distributed inference of GenAI applications (such as LLMs) on Private Cloud Compute. You will get to work on custom-built server hardware that brings the power and security of Apple silicon to the data center. Team also works on GPU acceleration of ML Training frameworks such as PyTorch and JAX using Metal runtime and device backend. We are looking for enginee
Posted 60+ days ago | Updated 4 hours ago

Java Software Architect

Wise Equation Solutions Inc.

•

Albany, New York, USA

Contract

Role: Expert Software Architect Location: Albany, NY 12210 Duration: 30 Months, Hybrid Role Roles and Responsibilities This individual will participate in the following activities: Implement Integrated Justice Portal standards related to application technologies, including UI standards, development methodology, and coding and style guides.Act as the primary technical liaison with technical infrastructure teams to resolve problems and implement solutions to technical issues impacting application
Easy Apply
Posted 11 days ago

ML Compute Acceleration Engineer

Apple, Inc.

•

No location provided

Full-time

Apple's Compute Frameworks team in GPU, Graphics and Displays org provides a suite of high-performance data parallel algorithms for developers inside and outside of Apple for iOS, macOS and Apple TV. Our efforts are currently focused in the key areas of linear algebra, image processing, machine learning, along with other projects of key interest to Apple. We are always looking for exceptionally dedicated individuals to grow our outstanding team to lay the foundation of technologies like Apple In
Posted 60+ days ago | Updated 4 hours ago

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 16 days ago | Updated 4 hours ago

Machine Learning Engineer, Efficiency Engineering - USDS

TikTok

•

San Jose, California, USA

Full-time

Location : San Jose Employment Type : Regular Job Code : A87619 Apply to this job Share this listing: Responsibilities The Efficiency Engineering team is all about our passion for crafting innovative tools and applications that empower IT operations and devops teams to achieve new levels of efficiency. We're a tight-knit crew of experienced developers, engineers and problem solvers fueled by a shared vision: streamlining operations, reducing manual workload, and empowering teams to do the
Posted 37 days ago | Updated 4 hours ago

Principal Machine Rendering Engineer

Epic Games

•

Cary, North Carolina, USA

Full-time

WHAT MAKES US EPIC? At the core of Epic's success are talented, passionate people. Epic prides itself on creating a collaborative, welcoming, and creative environment. Whether it's building award-winning games or crafting engine technology that enables others to make visually stunning interactive experiences, we're always innovating. Being Epic means being a part of a team that continually strives to do right by our community and users. We're constantly innovating to raise the bar of engine an
Posted 51 days ago | Updated 4 hours ago

Sr. Software Engineer TS/SCI Polygraph

Leidos

•

Annapolis, Maryland, USA

Full-time

Leidos has an exciting opportunity for a Sr. Software Engineer! *Must have an active TS/SCI Polygraph up front. No exceptions.* You will perform software development lifecycle (SDLC) activities as both an individual and a member of our top-notch agile development team building a large complex enterprise system. Development includes the full range of turning Agile user stories into implementable concepts, through development, unit testing, integration and test, and deployment of the new capabili
Posted 23 days ago | Updated 4 hours ago

Sr. Computer Vision & Machine Learning Engineer, Creativity Apps

Apple, Inc.

•

No location provided

Full-time

At Apple, new ideas have a way of becoming phenomenal products, services, and customer experiences very quickly! The Creativity Apps team needs your help shaping the next generation of creative editing tools by working on pioneering technologies to surprise and delight creative pros and enthusiasts alike. As Senior Computer Vision & Machine Learning Engineer, you will be working alongside our world-class creatives, designers, and engineers to help innovate in the creative space in ways that only
Posted 60+ days ago | Updated 4 hours ago

Staff AI/ML Engineer - Onboard Embodied AI

General Motors

•

Mountain View, California, USA

Full-time

Job Description Hybrid: This role is categorized as hybrid. This means the successful candidate is expected to report to the Mountain View Technical Center in the Bay Area three times per week, at minimum. Role: As a Technical Lead in Machine Learning within the Onboard Embodied AI organization, you will be a senior individual contributor driving cutting-edge end-to-end machine learning solutions directly impacting autonomous driving performance. Your role is pivotal in designing, architecting,
Posted 60+ days ago | Updated 1 hour ago

Senior Lead Software Engineer - Machine Learning Engineer

JPMorgan Chase & Co.

•

Jersey City, New Jersey, USA

Full-time

Job Description Be an integral part of an agile team that's constantly pushing the envelope to enhance, build, and deliver top-notch technology products. Are you ready to make a significant impact in the world of AI and machine learning? At JPMorgan Chase, in this role, you'll be a key player in an agile team dedicated to enhancing, building, and delivering market-leading technology products that are secure, stable, and scalable As a Senior Lead Software/Machine Learning Engineer, at JPMorgan
Posted 56 days ago | Updated 4 hours ago