--- title: JAX: Is This Google’s NumPy killer? | daily.dev description: JAX, developed by Google, aims to enhance numerical computing in Python by offering NumPy-like functionality with advanced features such as automatic... canonical: https://daily.dev/posts/jax-is-this-google-s-numpy-killer--1fuxvogzi twitter:card: summary_large_image twitter:site: @dailydotdev og:type: website og:site_name: daily.dev og:title: JAX: Is This Google’s NumPy killer? | daily.dev og:description: JAX, developed by Google, aims to enhance numerical computing in Python by offering NumPy-like functionality with advanced features such as automatic... og:url: https://daily.dev/posts/jax-is-this-google-s-numpy-killer--1fuxvogzi og:image: https://api.daily.dev/og/posts/1fUxVogZI.png og:image:alt: JAX: Is This Google’s NumPy killer? og:image:width: 1200 og:image:height: 630 og:locale: en --- > ## Documentation Index > Fetch the complete documentation index at: https://daily.dev/llms.txt > Use this file to discover all available pages before exploring further. # JAX: Is This Google’s NumPy killer? **[Towards Data Science](https://daily.dev/sources/tds)** · 17 min read · 0 upvotes · 0 comments ## Summary JAX, developed by Google, aims to enhance numerical computing in Python by offering NumPy-like functionality with advanced features such as automatic differentiation and JIT compilation. Unlike NumPy, JAX operates efficiently on GPUs and TPUs and supports asynchronous execution. While not yet an official Google product, JAX shows promising performance gains, especially in machine learning and large scientific simulations. It is not a complete replacement for NumPy, with some differences in execution backend and API coverage. JAX presents a compelling option for tasks with performance bottlenecks or requiring gradient computation. ## Full article daily.dev links to this article rather than hosting it. Read it at the original source: --- Tags: [#machine-learning](https://daily.dev/tags/machine-learning), [#python](https://daily.dev/tags/python), [#google](https://daily.dev/tags/google), [#gpu](https://daily.dev/tags/gpu) [View this post on daily.dev](https://daily.dev/posts/jax-is-this-google-s-numpy-killer--1fuxvogzi) ```json {"@context":"https://schema.org","@graph":[{"@type":"Organization","@id":"https://daily.dev/#organization","name":"daily.dev","url":"https://daily.dev","logo":{"@type":"ImageObject","url":"https://daily.dev/apple-touch-icon.png","width":180,"height":180},"sameAs":["https://twitter.com/dailydotdev","https://github.com/dailydotdev","https://www.linkedin.com/company/daily-dev-ltd"]},{"@type":"WebSite","@id":"https://daily.dev/#website","url":"https://daily.dev","name":"daily.dev","publisher":{"@id":"https://daily.dev/#organization"},"potentialAction":{"@type":"SearchAction","target":{"@type":"EntryPoint","urlTemplate":"https://daily.dev/search?q={search_term_string}"},"query-input":"required name=search_term_string"}}]} {"@context":"https://schema.org","@type":"TechArticle","headline":"JAX: Is This Google’s NumPy killer?","url":"https://daily.dev/posts/jax-is-this-google-s-numpy-killer--1fuxvogzi","mainEntityOfPage":{"@type":"WebPage","@id":"https://daily.dev/posts/jax-is-this-google-s-numpy-killer--1fuxvogzi"},"datePublished":"2025-05-28T20:07:32.175Z","dateModified":"2025-05-28T20:07:48.952Z","description":"JAX, developed by Google, aims to enhance numerical computing in Python by offering NumPy-like functionality with advanced features such as automatic...","image":"https://media.daily.dev/image/upload/f_auto,q_auto/v1/posts/138c649e4799a25d1c5da6d0271c7ba5?_a=AQAEuj9","thumbnailUrl":"https://media.daily.dev/image/upload/f_auto,q_auto/v1/posts/138c649e4799a25d1c5da6d0271c7ba5?_a=AQAEuj9","isAccessibleForFree":true,"articleSection":"Towards Data Science","inLanguage":"en","publisher":{"@type":"Organization","name":"daily.dev","url":"https://daily.dev","logo":{"@type":"ImageObject","url":"https://daily.dev/apple-touch-icon.png","width":180,"height":180}},"author":{"@type":"Organization","name":"Towards Data Science","logo":"https://media.daily.dev/image/upload/t_logo,f_auto/v1/logos/tds","url":"https://daily.dev/sources/tds"},"commentCount":0,"discussionUrl":"https://daily.dev/posts/jax-is-this-google-s-numpy-killer--1fuxvogzi","interactionStatistic":[{"@type":"InteractionCounter","interactionType":{"@type":"LikeAction"},"userInteractionCount":0},{"@type":"InteractionCounter","interactionType":{"@type":"CommentAction"},"userInteractionCount":0}],"keywords":"machine-learning,python,google,gpu","timeRequired":"PT17M"} {"@context":"https://schema.org","@type":"BreadcrumbList","itemListElement":[{"@type":"ListItem","position":1,"name":"Home","item":"https://daily.dev"},{"@type":"ListItem","position":2,"name":"Towards Data Science","item":"https://daily.dev/sources/tds"},{"@type":"ListItem","position":3,"name":"JAX: Is This Google’s NumPy killer?"}]} ```