"jax" Jobs
16 open tech roles matching “jax”, taken straight from company career pages — not reposted from other job boards. Most in demand right now: PyTorch, Python, JAX. Every listing is re-checked daily and closed roles are removed.
Showing 16 of 16 results
Develop and apply methods at the intersection of 3D molecular simulation and machine learning to impact drug discovery programs.
Join Fireworks AI as a Member of Technical Staff to advance generative AI through foundational research and collaboration with top experts.
Join Fireworks AI as a Member of Technical Staff to advance generative AI and multimodal systems through foundational research.
Develop and implement software solutions for robotic manipulation systems at Skild AI, focusing on state-of-the-art learning-based algorithms.
Join Skild AI as a Robotics Engineer to develop software solutions for robotic manipulation systems using state-of-the-art algorithms.
Join Skild AI as a Research Engineer to enhance and deploy robotic intelligence in real-world environments.
Join Skild AI as a Software Engineer to develop and optimize software infrastructure for training advanced AI models in robotics.
Develop and refine deep learning models for robotic manipulation at a pioneering AI startup.
Join Skild AI as a Machine Learning Engineer to develop cutting-edge reinforcement learning algorithms for robotic applications.
Join Skild AI as a Machine Learning Engineer to develop cutting-edge reinforcement learning algorithms for adaptable robotic systems.
Join Skild AI as a Research Engineer to develop scalable robotic systems for general-purpose robotic intelligence.
Join Skild AI as a Research Scientist to develop advanced techniques for robotic systems in a passionate and innovative team.