Google. NumPy on accelerators. Autograd.
Visit JAXJAX is an artificial intelligence tool designed to help users improve productivity and automate tasks using modern AI technology. AI tools like JAX are commonly used for content creation, research, coding, design and automation.