"jax" Jobs
106 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 106 results
Join Periodic Labs as a Computational Scientist to develop differentiable simulations for continuum-physics problems using AI.
Join Perplexity AI as a Search Machine Learning Engineer Intern to enhance search technologies in a collaborative environment.
Join Inferact as a co-op student to work on cutting-edge AI inference systems in a hands-on engineering role.
Lead the unification of large datasets to power generative audio models in a fast-moving team at Udio.
Join Perplexity AI as a Search Machine Learning Engineer Intern to enhance search technologies in a collaborative environment.
Join CoreWeave as a Senior Applied ML Engineer to tackle challenges in continuous learning for AI agents with a focus on innovative solutions.
Join Applied Intuition as an ML Runtime Optimization Engineer to optimize ML models for embedded environments in a collaborative team.
Join AssemblyAI as a Senior Research Engineer to enhance large-scale distributed training and inference systems in Voice AI.
Join Periodic Labs as an ML Systems Engineer to build and optimize large-scale training and reinforcement learning infrastructure.
Join Baseten as a Forward Deployed Engineer to solve complex AI challenges for leading companies.
Develop and apply methods at the intersection of 3D molecular simulation and machine learning to impact drug discovery programs.
Join CoreWeave as a Senior Engineer to optimize GPU kernels for high-performance AI applications in a rapidly growing environment.
Join Fireworks AI as a Member of Technical Staff to advance generative AI through foundational research and collaboration with top experts.
Related searches