--- title: nnetsauce with and without jax for GPU acceleration description: nnetsauce version 0.51.2 removes jax and jaxlib from its default installation due to the heavy size of jaxlib. Users who want GPU acceleration must now... canonical: https://daily.dev/posts/nnetsauce-with-and-without-jax-for-gpu-acceleration-bxjilivnf twitter:card: summary_large_image twitter:site: @dailydotdev og:type: website og:site_name: daily.dev og:title: nnetsauce with and without jax for GPU acceleration | daily.dev og:description: nnetsauce version 0.51.2 removes jax and jaxlib from its default installation due to the heavy size of jaxlib. Users who want GPU acceleration must now... og:url: https://daily.dev/posts/nnetsauce-with-and-without-jax-for-gpu-acceleration-bxjilivnf og:image: https://api.daily.dev/og/posts/BXjilivnF.png og:image:alt: nnetsauce with and without jax for GPU acceleration 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. # nnetsauce with and without jax for GPU acceleration **[R-bloggers](https://daily.dev/sources/rbloggers)** · 1 min read · 0 upvotes · 0 comments ## Summary nnetsauce version 0.51.2 removes jax and jaxlib from its default installation due to the heavy size of jaxlib. Users who want GPU acceleration must now explicitly install jax as an optional dependency using pip, uv pip, or conda commands. ## Full article daily.dev links to this article rather than hosting it. Read it at the original source: ## Similar posts on daily.dev - [Wheel Next \+ Packaging PEPs](https://daily.dev/posts/wheel-next-packaging-peps-z6njzzzu4) · Planet Python · 0 upvotes · 0 comments - [Run High-Performance Core Math at Scale with NVIDIA nvmath-python](https://daily.dev/posts/run-high-performance-core-math-at-scale-with-nvidia-nvmath-python-gytxensnb) · NVIDIA Developer · 0 upvotes · 0 comments - [Building production AI on Google Cloud TPUs with JAX](https://daily.dev/posts/building-production-ai-on-google-cloud-tpus-with-jax-sqx7nfdgl) · Google Developers · 1 upvotes · 0 comments - [How JAX makes high-performance economics accessible](https://daily.dev/posts/how-jax-makes-high-performance-economics-accessible-spechnovm) · Google Open Source Blog · 0 upvotes · 0 comments --- Tags: [#machine-learning](https://daily.dev/tags/machine-learning), [#python](https://daily.dev/tags/python), [#gpu](https://daily.dev/tags/gpu) [View this post on daily.dev](https://daily.dev/posts/nnetsauce-with-and-without-jax-for-gpu-acceleration-bxjilivnf) ```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":"nnetsauce with and without jax for GPU acceleration","url":"https://daily.dev/posts/nnetsauce-with-and-without-jax-for-gpu-acceleration-bxjilivnf","mainEntityOfPage":{"@type":"WebPage","@id":"https://daily.dev/posts/nnetsauce-with-and-without-jax-for-gpu-acceleration-bxjilivnf"},"datePublished":"2026-03-02T17:42:20.583Z","dateModified":"2026-03-02T17:43:00.304Z","description":"nnetsauce version 0.51.2 removes jax and jaxlib from its default installation due to the heavy size of jaxlib. Users who want GPU acceleration must now...","image":"https://media.daily.dev/image/upload/f_auto,q_auto/v1/posts/e464a8b9a77b72ff27fb5b45ddf5a3bf?_a=AQAEuop","thumbnailUrl":"https://media.daily.dev/image/upload/f_auto,q_auto/v1/posts/e464a8b9a77b72ff27fb5b45ddf5a3bf?_a=AQAEuop","isAccessibleForFree":true,"articleSection":"R-bloggers","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":"R-bloggers","logo":"https://media.daily.dev/image/upload/t_logo,f_auto/v1/logos/98bcbbdf72774292a55ceda8f2277f1e","url":"https://daily.dev/sources/rbloggers"},"commentCount":0,"discussionUrl":"https://daily.dev/posts/nnetsauce-with-and-without-jax-for-gpu-acceleration-bxjilivnf","interactionStatistic":[{"@type":"InteractionCounter","interactionType":{"@type":"LikeAction"},"userInteractionCount":0},{"@type":"InteractionCounter","interactionType":{"@type":"CommentAction"},"userInteractionCount":0}],"keywords":"machine-learning,python,gpu","timeRequired":"PT1M"} {"@context":"https://schema.org","@type":"BreadcrumbList","itemListElement":[{"@type":"ListItem","position":1,"name":"Home","item":"https://daily.dev"},{"@type":"ListItem","position":2,"name":"R-bloggers","item":"https://daily.dev/sources/rbloggers"},{"@type":"ListItem","position":3,"name":"nnetsauce with and without jax for GPU acceleration"}]} ```