Jax Jobs in United States

Join Applied Intuition as an ML Runtime Optimization Engineer to optimize ML models for embedded environments in a collaborative team.

Join P-1 AI as an AI Research Scientist to push the boundaries of AI applied to the physical world in a hands-on role.

Join Krea as an ML Engineer to build personalization and recommendation systems that enhance creative AI tools.

Join Atoms as a Staff Machine Learning Engineer to bridge AI research and real-world physical actuation in autonomous transport.

Join Atoms as a Senior Machine Learning Engineer to bridge AI research and real-world physical actuation in autonomous transport.

Develop and apply methods at the intersection of 3D molecular simulation and machine learning to impact drug discovery programs.

Join Freenome as a Staff Machine Learning Scientist to drive innovative AI research in cancer detection.

Join Abridge as a Machine Learning Scientist to advance medical NLP and contribute to impactful healthcare solutions.

Lead the design and development of core ML models for Instacart’s ads ecosystem in a fully remote role.

Join CoreWeave as a Senior Engineer to optimize GPU kernels for high-performance AI applications in a rapidly growing 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 Cloudflare as a Senior Machine Learning Engineer to optimize and productionize ML models for a global serverless inference platform.

Join Block as an Applied Research Intern to develop proactive intelligence systems that enhance customer understanding and decision-making.