JAX jobs

Senior Software Development Engineer - AI/ML, AWS Neuron, Multimodal Inference

Seattle, Washington

•

Today

The Annapurna Labs team at Amazon Web Services (AWS) builds AWS Neuron, the software development kit used to accelerate deep learning and GenAI workloads on Amazon?s custom machine learning accelerators, Inferentia and Trainium.The AWS Neuron SDK, developed by the Annapurna Labs team at AWS, is the backbone for accelerating deep learning and GenAI workloads on Amazon's Inferentia and Trainium ML accelerators. This comprehensive toolkit includes an ML compiler, runtime, and application framework

Full-time

Compensation information provided in the description

Principal Software Development Engineer, ML Training and Performance

San Jose, California

•

Today

WHAT YOU DO AT AMD CHANGES EVERYTHING At AMD, our mission is to build great products that accelerate next-generation computing experiences-from AI and data centers, to PCs, gaming and embedded systems. Grounded in a culture of innovation and collaboration, we believe real progress comes from bold ideas, human ingenuity and a shared passion to create something extraordinary. When you join AMD, you'll discover the real differentiator is our culture. We push the limits of innovation to solve the wo

Full-time

USD 198,100.00 per year

Java Architect - HBITS-07-14493

Albany, New York

•

Today

84 months of experience developing highly available and scalable multitier applications using Object Oriented or Service Oriented Architecture involving data structures, algorithms, software design and JEE Design Patterns. 84 months of experience developing applications that are implemented with JAVA Development Kit (JDK 1.5 or higher). 84 months of experience performing development of applications developed using JPA and/or Hibernate using Oracle or SQL Server databases 84 months of experience

Easy Apply

Contract, Third Party

Java Full Stack Developer -VP

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

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 Software Engineer, Robotics - Isaac Lab

Santa Clara, California

•

Today

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

Full-time

USD 148,000.00 per year

Lead Forward Deployed Software Engineer (1)

Santa Clara, California

•

Today

Full-time

USD 198,100.00 per year

Senior Reinforcement Learning Engineer

Austin, Texas

•

Today

Apptronik is a human-centered robotics company developing AI-powered robots to support humanity in every facet of life. Our flagship humanoid robot, Apollo, is built to collaborate thoughtfully with people, starting with critical industries such as manufacturing and logistics, with future applications in healthcare, the home, and beyond. We operate at the cutting edge of embodied AI, applying our expertise across the full robotics stack to solve some of society's most important problems. You wi

Full-time

IT Technical Specialist

Columbus, Ohio

•

Today

Description In this position, the successful candidate will design and implement complex IT solutions that are in line with the enterprise architecture roadmap. The successful candidate will: Deliver well-designed, quality software in either an agile or waterfall environmentWork on a team to ensure successful delivery of solutions that add value to Huntington's colleagues and customers as-well-as achieve business objectives.Understand business requirements.Communicate clearly with business and

Full-time

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

2026 PhD Graduate - Artificial Intelligence and Machine Learning (AI/ML) Research Scientist

Laurel, Maryland

•

Today

Description Are you an early-career researcher hoping to invent the future of artificial intelligence (AI)? Do you have a burning desire for your research to serve a bigger purpose? Are you concerned about pressing challenges facing our nation's security and prosperity? If these questions describe you, you may be the kind of person the Artificial Intelligence Group at JHU/APL is looking for. We are a team within the Research and Exploratory Development Department (REDD) that strives to close

Full-time

USD 102,500.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 GenAI Algorithms Engineer - Model Optimizations for Inference

Santa Clara, California

•

Today

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 quantization, speculative decoding, sparsity, distillation, pruning to neural architecture search, and streamlined deployment strategies with open-sourced inference frameworks. Seeking a Senior Deep Learning Algorithms Engin

Full-time

USD 148,000.00 per year

Senior Staff Software Development Engineer- GPU/AI/ML

Santa Clara, California

•

Today

Full-time

USD 168,000.00 per year

Principal GPU Performance Engineer - Artificial Intelligence

Austin, Texas

•

Today

Full-time

USD 175,000.00 per year

Java Architect w/ Criminal Justice Exp - HBITS-07-14305

Albany, New York

•

Today

84 months experience designing and implementing highly available and scalable multitier applications using Object Oriented or Service Oriented Architecture with experience in data structures, algorithms, software design and JEE Design Patterns on an IBM platform. 84 months experience developing, implementing and integrating systems related to criminal justice processing at the state or federal level (including but not limited to: disposition processing, fingerprint identification processing, cri

Easy Apply

Third Party, Contract

Senior Staff Software Development Engineer- GPU, LLM, AI

Santa Clara, California

•

Today

Full-time

USD 168,000.00 per year

Senior Java JEE Developer

Hybrid in Albany, New York

•

16d ago

Job Title: Senior Java JEE Developer(WebSphere 9, JSF/JSP, JMS/EJB/MQ, Hibernate/Oracle) Location: Albany, NY (Hybrid) We are currently seeking candidates who meet the following qualification Qualifications: Proven experience leading iterative development teams through the planning, design, implementation, and integration of complex JEE-based systems, with strong knowledge of data structures, algorithms, and software design principles. Hands-on experience developing applications using JDK 1.8+

Easy Apply

Contract

Depends on Experience

Machin Learning Engineer

Fremont, California

•

21d ago

Description: Serve as a member of the factory software machine learning and computer vision team. Design, develop and implement critical machine learning models that operate on our factory and warehouse environments. Take ambiguous problem statements and build end-to-end models using a variety of techniques and tools like supervised learning, convolutional neural networks, Pytorch, Pandas, etc. Work closely with partners in production, process, controls, quality, etc. to build solutions for the

Easy Apply

Contract, Third Party

$60 - $70

Machine Learning Engineer

Fremont, California

•

8d ago

Title- Machine Learning Engineer Location- Fremont, CA - 5 Days Onsite/Week Type- Contract- Duration- 6+ Months Mode of Interview- F2F Job Description: Serve as a member of the factory software machine learning and computer vision team. Design, develop and implement critical machine learning models that operate on our factory and warehouse environments. Take ambiguous problem statements and build end-to-end models using a variety of techniques and tools like supervised learning, convolutional

Easy Apply

Third Party, Contract

Depends on Experience

Filter Results

Job post features

Posted date

Work settings

Employment type

Distance

Employer type

Work authorization