"jax" Jobs
38 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 38 results
Join Periodic Labs as a Computational Scientist to develop differentiable simulations for continuum-physics problems using AI.
Lead the unification of large datasets to power generative audio models in a fast-moving team at Udio.
Join CoreWeave as a Senior Applied ML Engineer to tackle challenges in continuous learning for AI agents with a focus on innovative solutions.
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.
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 DeepL as a Senior Research Scientist to innovate in reinforcement learning and shape the future of AI technology.
Join SpaceX as a Senior Software Engineer to develop AI systems that enhance engineering processes for launch vehicles and spacecraft.
Drive the development of AI-powered clinician tools at Abridge, combining clinical expertise with AI in a hybrid work environment.
Join Cloudflare as a Senior Machine Learning Engineer to optimize and productionize ML models for a global serverless inference platform.
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 Xaira Therapeutics as a Senior Software Engineer to build AI infrastructure for drug discovery and development.
Related searches