Jax Jobs
Browse 20 exciting jobs hiring in Jax now. Check out companies hiring such as humans& ai, inc, NVIDIA, Visa in Plano, Glendale, Indianapolis.
Design and optimize the distributed systems and numerical foundations that enable large-scale RL and model training at humans&.
NVIDIA is hiring a Senior Software Engineer, ML to implement and optimize state-of-the-art machine learning models across multiple frameworks and cutting-edge GPU hardware to maximize performance.
Lead Visa Research's generative AI strategy and team to advance cutting-edge GenAI models, agents, and privacy techniques and deliver real-world impact across payments and commerce.
Lead the design and deployment of de novo design and closed-loop ML frameworks to accelerate molecular engineering and experimental discovery at Merge Labs.
Work at the intersection of research and engineering to run, debug, and improve large-scale training and reinforcement learning methods for cutting-edge language models at humans&.
Sciforium is hiring a Distributed Training Engineer to own and optimize the full ML training stack — from drivers and kernels to JAX/PyTorch — enabling large-scale training and deployment of next-generation LLMs.
An ML Researcher role at Prima Mente to develop and evaluate large-scale biological foundation models that translate into real clinical and mechanistic insights across neuroscience and complex disease.
Lead the ML stack as a founding Machine Learning Engineer at a stealth, self-funded AI group, defining models, training pipelines, and scalable inference for a global consumer product.
Contribute as an ML/AI intern at Oligo to build generative-design, simulation-aware, and systems-level AI that helps design and produce flight-ready spacecraft.
A founding AI/ML research engineer role to design and build core model, data, and inference systems for a stealth, high-impact consumer AI product backed by a profitable US$2B group.
Lead the technical design and implementation of A1’s foundational LLM systems—training pipelines, inference stacks, and deployment architecture—for a global consumer AI product.
A six-month, full-time Research Residency at OpenAI in San Francisco for technical builders and researchers who want hands-on mentorship and the opportunity to rapidly contribute to frontier AI research.
Senior ML Systems Engineer to own and evolve the training framework and tooling that enables reliable, high-performance large-scale LLM training.
Be a founding deep learning researcher building and deploying foundation models for physics in a fast-paced, in-person San Francisco startup working directly with the CEO and founding team.
Lead the design and implementation of compiler, runtime, and debugger integrations for PyTorch, TensorFlow, JAX, and MXNet on custom hardware to maximize AI model performance at Flux.
Basis is hiring an experienced ML Systems Engineer to build, operate, and optimize distributed training and cloud infrastructure that enables large-scale, reproducible AI research.
Lead the design and operation of multi-cloud GPU/TPU HPC infrastructure to enable scalable, high-performance distributed training for cutting-edge AI research and products.
Cohere is hiring a Staff Software Engineer to build and operate ML-optimized HPC infrastructure (Kubernetes-based GPU/TPU superclusters) that accelerates research and production training of large AI models.
Ignite IT is hiring a US-citizen Java/Angular Full Stack Developer to modernize DHS enterprise systems within an Agile team.
OpenBlock Labs is hiring a Research Engineer to develop reinforcement learning and continual learning systems that enable coding agents to learn and adapt from developer feedback and real codebases.
How much do jax jobs pay?
Below 50k*
0
|
50k-100k*
1
|
Over 100k*
19
|
Top companies hiring for jax jobs
- Bjak 3
- humans& ai, inc 2
- Awesome Motive 2
- Cohere 2
- Gauss Labs 1
- NVIDIA 1
- Visa 1
Best cities to find jax jobs
- San Francisco 6
- New York 3
- Palo Alto 1
- Hawthorne 1
