"jax" Jobs
17 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 17 of 17 results
Join Fireworks AI as a Member of Technical Staff to advance generative AI through foundational research and collaboration with top experts.
Join Perplexity AI as an AI Inference Engineer to optimize and develop our inference engine for various model architectures.
Join Udio as a full-stack scientist to lead quantitative research efforts at the intersection of music and AI.
Join Perplexity AI as a Technical Staff member to enhance our AI inference engine with cutting-edge technologies.
Lead the development of machine learning initiatives to enhance customer support experiences at Block.
Join Reflection AI as a Research Software Engineer to bridge research and production in cutting-edge AI training systems.
Join Fireworks AI as a Member of Technical Staff to advance generative AI and multimodal systems through foundational research.
Join Preference Model as a Research Engineer to advance self-directed learning in machine learning with a focus on RL environments.
Join Preference Model as a Research Engineer to advance self-directed learning in large language models within a fast-paced startup.
Join Arena as a Machine Learning Scientist to evaluate AI models and contribute to impactful research in a collaborative environment.
Join Anthropic as a Staff Software Engineer to build scalable ML infrastructure for AI safety systems.
Join Freenome as a Staff Machine Learning Scientist to drive innovative AI research in cancer detection.
Related searches