JAX
123 jobs tagged JAX
JAX jobs — quick facts
- How many JAX jobs are available?
- 123 active JAX roles sourced directly from company career pages, updated daily on Joblaze.
- What is the median salary for JAX jobs?
- The median advertised salary for JAX roles on Joblaze is about $208,000 per year.
- Which companies are hiring for JAX?
- Companies currently hiring JAX include Graphcore, Anthropic, Skild AI, Together AI, Perplexity.
- Are JAX jobs remote?
- 4% of active JAX roles on Joblaze offer remote work.
Join Profound as a Post-Training Engineer to enhance AI models for Fortune 500 marketing teams in a fast-paced environment.
Join Harvey as a Research Engineer to build infrastructure for ML models in a fast-paced, fully remote environment.
Join Harvey AI as a Member of Technical Staff to build infrastructure for ML models in legal work.
Join Atoms as an AI Research Intern to bridge AI research and real-world applications in autonomous transport platforms.
Join Skydio as an intern to develop reinforcement learning methods for autonomous drones in real-world applications.
Own and improve the development experience by designing and maintaining build systems and developer tooling at Eka Robotics.
Join Eka Robotics as a Machine Learning Engineer to develop cutting-edge algorithms for robotic manipulation in a fast-paced startup environment.
Join Eka Robotics as a Machine Learning Engineer to develop cutting-edge computer vision systems for robotic manipulation.
Join Eka Robotics as a Machine Learning Infrastructure Engineer to build and maintain large-scale model training systems for robotics.
Join K2 Space as a Photonic IC Design Engineer to lead photonic design automation and modeling for cutting-edge satellite technology.
Join Graphcore as a Software Engineer to enhance AI frameworks and optimize performance for machine learning workloads.
Join Reflection AI as a Member of Technical Staff to design and implement data sourcing and curation techniques for AI models.
Join Atoms as an AI/ML Research Intern to bridge high-level AI research with real-world physical actuation for autonomous transport platforms.
Lead data science initiatives to enhance user engagement and ecosystem understanding at Reddit.
Join Preference Model as a Research Engineer to design and build RL environments for advancing machine learning capabilities.
Join Preference Model as a Senior ML Infrastructure Engineer to build scalable infrastructure for post-training research on large language models.
Join Preference Model as a Research Engineer to advance self-directed learning in large language models within a fast-paced startup.
Join Graphcore as a Graduate Machine Learning Engineer and gain hands-on experience in AI systems over a two-year program.
Join Graphcore as a Graduate Software Engineer to work on AI/ML Test Systems with hands-on learning and development opportunities.
Join Mecka AI as a Research Scientist to advance multimodal foundation models and tackle challenging research problems.
Join Graphcore as a Senior Machine Learning Engineer to advance AI models on cutting-edge hardware in a hybrid work environment.
Join Graphcore as a Senior Machine Learning Engineer to advance AI models on cutting-edge hardware in a hybrid work environment.
Join Graphcore as a Senior Machine Learning Engineer to advance AI models on cutting-edge hardware in a hybrid work environment.
Join Lyft as a PhD Machine Learning Software Engineer Intern to tackle open research problems impacting millions of riders.
Join Anthropic as a Research Engineer/Scientist to develop next-generation RL algorithms and architectures at frontier scale.
Join Chai Discovery as a Research Engineer to optimize ML infrastructure for drug discovery in a fast-paced, innovative environment.
Join Discord as a Staff Machine Learning Engineer to build ML systems for consumer revenue across various platforms.
Join Together AI as a Research Intern to work on cutting-edge distributed inference and optimization for large foundation models.
Join Together AI as a Research Intern to explore frontier AI systems and contribute to innovative projects in agentic AI.
Join Together AI as a Research Intern to work on advanced machine learning techniques and contribute to innovative model shaping projects.
Join Together AI as a Research Intern to work on cutting-edge agentic AI systems in a collaborative environment.
Join Together AI as a Research Intern to work on cutting-edge distributed inference and optimization for large foundation models.
Join Together AI as a Research Intern to work on advanced machine learning techniques and contribute to innovative model shaping projects.
Join Proxima Fusion as a Software Engineer to develop next-generation simulation tools for stellarator design and research.
Own the strategy and execution for infrastructure-native products at the intersection of W&B and CoreWeave.
Join Discord as a Senior Machine Learning Engineer to build and deploy models that enhance user safety on the platform.
Join Inferact as a co-op student to work on cutting-edge AI inference systems in a hands-on engineering role.
Join Anthropic as a Staff Software Engineer to build scalable ML infrastructure for AI safety systems.
Join Fireworks AI as a Member of Technical Staff to advance generative AI and multimodal systems through foundational research.
Join Cantina as a Research Intern to work on next-gen video models in a hands-on, mentorship-driven environment.
Join Periodic Labs as a Computational Scientist to develop differentiable simulations for continuum-physics problems using AI.
Join ChipAgents as a Research Scientist to advance AI-assisted electronic design automation in a high-impact, mission-driven team.
Join Xaira Therapeutics as an AI Scientist to develop multimodal foundation models for drug discovery using advanced AI techniques.