Jax Jobs
Browse 47 exciting jobs hiring in Jax now. Check out companies hiring such as NVIDIA, Jobgether, Cobot in Fort Wayne, Dallas, New York.
NVIDIA seeks a Senior Deep Learning Performance Architect to design, model, and analyze next-generation AI hardware and software for optimal performance, cost, and power efficiency.
Lead the architecture and production of large-scale recommender and personalization systems as a Staff Software Engineer, Machine Learning at a remote-first US company focused on high-impact ML products.
Cobot is hiring an AI Research Engineer to develop and deploy transformer, diffusion, and reinforcement learning models that advance autonomous systems and human-robot interaction.
Technical Marketing Engineer needed to produce developer-focused technical content, samples, and benchmarks that demonstrate and improve NVIDIA's AI platform software usability.
Work on cutting-edge R&D to design and deploy LLMs, multimodal AI, and cognitive systems that improve human-AI collaboration for mission-driven projects.
NVIDIA is hiring a DGX Cloud Performance Engineer to lead performance benchmarking, modeling, and architecture design for large-scale accelerator-based AI clusters.
NVIDIA's MARS team is hiring a Senior AI/ML Storage Engineer to architect and operate exascale storage and orchestration systems powering frontier AI research and global-scale workloads.
Work remotely on the founding team to design, deploy, and optimize production ML systems that bridge applied research and real-world product impact.
Lead product strategy and execution for NVIDIA's AI infrastructure platform, driving roadmap, cross-functional alignment, and delivery of large-scale AI/ML and HPC solutions.
Lead the design and deployment of safety-first, aligned AI systems—combining research, engineering, and governance to ensure responsible autonomous agent behavior at scale.
Lead the design and delivery of adaptive, human-aware AI systems that improve trust and collaboration across text, speech, and interactive interfaces for a US-based organization.
Work with experienced engineers and researchers at DatologyAI to prototype and productionize ML systems that make model training faster, cheaper, and smarter.
Lead development of production-ready autonomous AI agents that set, decompose, and pursue long-horizon goals while ensuring safety and alignment.
Impact the next generation of AI-augmented coaching by building, fine-tuning, and shipping LLM-driven features and production ML systems at BetterUp.
Lead post-training and RL efforts to enable frontier models to design, evaluate, and autonomously run scientific experiments in a real lab environment at a fast-growing AI + physical sciences lab.
Lead the design and deployment of electromagnetic foundation models that combine neural operators with high-fidelity EM solvers to power Atlas, Arena’s industrial AI.
Trail of Bits is hiring a Machine Learning Security Research Fellow to perform original adversarial research on frontier AI/ML systems, develop novel attacks and defenses, and publish open-source tools and papers.
Lead development of state-of-the-art multimodal and video foundation models and drive their production integration for Eluvio's decentralized Content AI platform.
Contribute as a hands-on intern to build and optimize GPU-driven AI infrastructure and inference systems with a small engineering team in San Francisco.
Work as an integral engineering intern on GPU optimization, AI infrastructure, and inference systems to help design and implement performance-critical GPU tooling and architectures.
Lead the design and implementation of GPU-optimized infrastructure and systems to accelerate large model training and inference for a fast-moving AI infrastructure team.
Contribute to next-generation GPU optimization and AI infrastructure as a Spring intern, working hands-on with CUDA, deep learning frameworks, and systems engineering to accelerate inference and training.
A hands-on Perception Intern role at Mach9 to prototype and improve 2D/3D detection models for urban mapping and help transition algorithms to real-time edge systems.
Work on GPU-accelerated neural network software for real-time DNA sequencing analysis, optimizing C++/CUDA pipelines and collaborating with cross-functional teams to deliver high-performance diagnostic solutions.
PhD candidates in machine learning or related fields are invited to join Simular as a research intern to advance planning, multimodal grounding, reward modeling, and user intent understanding through hands-on experimentation and publication-quality research.
Lead the design and deployment of production-ready AI and ML models for multi-omics and clinical applications in a remote-first team advancing precision medicine.
Experienced deep learning engineer needed to develop, optimize, and deploy perception DNNs for autonomous vehicle platforms, improving inference speed, accuracy, and efficiency.
Work on state-of-the-art end-to-end ML systems for autonomous driving at Woven by Toyota, spanning data strategy, model design, experimentation, and production deployment.
Lead the design and execution of large-scale foundation model training and post-training pipelines, building performant distributed systems and custom kernels to deliver production-ready models.
Lead the design and deployment of cutting-edge ML models for an end-to-end autonomous driving stack at Woven by Toyota, working across teams to advance perception, prediction and planning.
Work on cutting-edge ML models and end-to-end learned autonomous driving systems at Woven by Toyota, collaborating across research and engineering teams to advance perception, prediction, and motion planning.
NVIDIA seeks a Senior Research Engineer to design, implement, and scale open-source post-training and RL algorithms for Nemotron generative AI models.
Lovelace AI seeks a Machine Learning Engineer to develop, optimize, and deploy high-performance ML models for national security and emergency-response applications.
Cohere is hiring PhD-level Research Interns to contribute to cutting-edge LLM research and help bring novel models and evaluations into production across multiple research areas.
NVIDIA is hiring an AI/ML Solutions Architect to partner with hyperscalers and cloud providers to design, optimize, and deploy large-scale AI training, inference, and infrastructure solutions.
Experienced research engineers are wanted to build and scale multimodal AI and reinforcement learning training systems for autonomous vehicles at NVIDIA, combining publishable research with production-grade GPU infrastructure work.
Lead original security research on state-of-the-art machine learning systems and work with top AI organizations to identify and mitigate novel attack vectors.
NVIDIA seeks an experienced Senior Director of Product Management to lead product strategy and execution for AI infrastructure platforms that power large-scale ML and HPC workloads.
Handshake AI is hiring a Summer 2026 PhD Research Intern to lead a focused LLM research project remotely, with mentorship and the goal of producing an archive-ready manuscript or top-tier conference submission.
Work on cutting-edge perception models and multimodal sensor fusion to power Seneca's autonomous firefighting drone systems as a core member of the engineering team.
Lead the development of scalable compute infrastructure for robotics foundation-model research at NVIDIA, optimizing GPU clusters, job orchestration, and observability for large training and evaluation workloads.
Sanofi is hiring a Senior Scientist, Systems Biology in Cambridge, MA to lead ML/AI development and multi-modal analytical pipelines that predict perturbations and functional synergy to accelerate drug discovery.
Lead technical engagements with top customers to architect, benchmark, and optimize large-scale AI and HPC solutions using NVIDIA GPU platforms.
NVIDIA seeks a Senior AI Software Engineer to extend Megatron Core and NeMo frameworks through distributed training innovations, performance tuning, and scalable tooling for large-scale LLM and multimodal model workflows.
Lead the design and operation of large-scale data infrastructure for NVIDIA's robotics foundation-model efforts (Project GR00T) to enable multimodal training and robot learning research.
Experienced Java and Angular full-stack developer needed to modernize and support DHS-facing enterprise applications in an Agile environment with a focus on secure, service-oriented solutions.
Lead the design and optimization of high-performance deep learning inference software on NVIDIA GPUs as a Senior Software Engineer on the TensorRT team.
How much do jax jobs pay?
Below 50k*
0
|
50k-100k*
4
|
Over 100k*
41
|
Top companies hiring for jax jobs
- NVIDIA 13
- Jobgether 7
- Herdora 4
- Woven by Toyota 3
- Trail of Bits 2
- Cobot 1
- Handshake 1
Best cities to find jax jobs
- San Francisco 5
- Seattle 1
- Remote 1
- Redwood City 1
