"jax" Jobs
66 open tech roles matching “jax”, taken straight from company career pages — not reposted from other job boards. Most in demand right now: JAX, PyTorch, Python. Every listing is re-checked daily and closed roles are removed.
Showing 20 of 66 results
Join Together AI as a Research Intern to work on cutting-edge distributed inference and optimization for large foundation models.
Join Cloudflare as a Senior Machine Learning Engineer to optimize and productionize ML models for a global serverless inference platform.
Join Atoms as an AI/ML Research Intern to bridge high-level AI research with real-world physical actuation for autonomous transport platforms.
Own the strategy and execution for infrastructure-native products at the intersection of W&B and CoreWeave.
Join Block as an Applied Research Intern to develop proactive intelligence systems that enhance customer understanding and decision-making.
Join Reflection AI as a Member of Technical Staff to design and implement data sourcing and curation techniques for AI models.
Join Chai Discovery as a Research Engineer to optimize ML infrastructure for drug discovery in a fast-paced, innovative environment.
Lead the design and development of core ML models for Instacart’s ads ecosystem in a fully remote role.
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 explore frontier AI systems and contribute to innovative projects in agentic AI.
Join Xaira Therapeutics as a Senior Software Engineer to build AI infrastructure for drug discovery and development.
Lead the development of machine learning initiatives to enhance customer support experiences at Block.
Join Inductive Bio as a Machine Learning expert to innovate on drug discovery models and collaborate with scientists.