"jax" Jobs
110 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 110 results
Join Graphcore's Triton team to enhance AI frameworks and optimize software for cutting-edge machine learning workloads.
Lead the design and development of core ML models for Instacart’s ads ecosystem in a fully remote role.
Join World Labs as a Performance Engineer to optimize AI models for speed and efficiency in a cutting-edge research environment.
Join Netic as an Applied AI Research Engineer to drive impactful ML projects in a cutting-edge AI-first environment.
Join Graphcore's Triton team to enhance AI frameworks and improve performance on cutting-edge hardware.
Join Ideogram as an Applied ML Engineer to bridge research and product, turning generative models into production features.
Join Atomic AI as a Scientist to develop machine learning models for RNA structure prediction and drug targeting.
Join Xaira Therapeutics as a Senior Software Engineer to build AI infrastructure for drug discovery and development.
Join Udio as a full-stack scientist to lead quantitative research efforts at the intersection of music and AI.
Join Cantina as a Research Intern to work on next-gen video models in a hands-on, mentorship-driven environment.
Join Twilio as a Machine Learning Engineer to design AI-powered features that enhance customer conversations.
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 ChipAgents as a Research Scientist to advance AI-assisted electronic design automation in a high-impact, mission-driven team.
Join Discord as a Senior Machine Learning Engineer to build and deploy models that enhance user safety on the platform.
Related searches