{"id":8625,"date":"2020-02-17T11:36:45","date_gmt":"2020-02-17T16:36:45","guid":{"rendered":"https:\/\/g33kinfo.com\/info\/?p=8625"},"modified":"2020-02-17T11:36:47","modified_gmt":"2020-02-17T16:36:47","slug":"trax-your-path-to-advanced-deep-learning","status":"publish","type":"post","link":"https:\/\/g33kinfo.com\/info\/trax-your-path-to-advanced-deep-learning\/","title":{"rendered":"Trax: your path to advanced deep learning"},"content":{"rendered":"<div class=\"github-embed github-embed-repository github-logo-mark\">\n<p>        <a href=\"https:\/\/github.com\/google\/trax\" target=\"_blank\">\t\t\t<strong>\t\t\t\tTrax \u2014 Deep Learning with Clear Code and Speed\t\t\t<\/strong>\t\t<\/a>\t\t<br \/>        <a href=\"https:\/\/github.com\/google\/trax\" target=\"_blank\">https:\/\/github.com\/google\/trax<\/a><br \/>        <a href=\"https:\/\/github.com\/google\/trax\/network\" target=\"_blank\">823<\/a> forks.<br \/>        <a href=\"https:\/\/github.com\/google\/trax\/stargazers\" target=\"_blank\">8,308<\/a> stars.<br \/>        <a href=\"https:\/\/github.com\/google\/trax\/issues\" target=\"_blank\">125<\/a> open issues.        <\/p>\n<details open>\n<summary>Recent commits:<\/summary>\n<ul class=\"github_commits\">\n<li class=\"github_commit\">                        <a href=\"https:\/\/github.com\/google\/trax\/commit\/31022d6cd7dd525ed11a04d84cd3936228499173\" target=\"_blank\">[trax] Explicitly set `jax_pmap_shmap_merge=False`.`trainer._multi_device_update_fn` uses `jax.pmap` and when `jax_pmap_shmap_merge=True`, `jax.pmap` requires inputs be explicitly sharded as the underlying `jax.jit` expects.This would need to be fixed if `jax_pmap_shmap_merge=True`.PiperOrigin-RevId: 811810947<\/a>, Copybara-Service                    <\/li>\n<li class=\"github_commit\">                        <a href=\"https:\/\/github.com\/google\/trax\/commit\/3d4a2760e6308627017c12a4892ef80c8d5a9a91\" target=\"_blank\">Remove more dtype canonicalization.Notable changes:* Remove dtype canonicalization from many lax dtype rules. Dtype rules should always talk about post-canonicalized types.* Use `check_and_canonicalize_user_dtype` instead of `canonicalize_dtype` in many user-callable lax functions.* Change some calls to lax functions to pass canonicalized types to avoid new user warnings.* Move `_convert_and_clip_integer` into `random.py`, which is its only user.* In passing, rename `_input_dtype` to `input_dtype` because it is exported from its enclosing module.PiperOrigin-RevId: 800201348<\/a>, Copybara-Service                    <\/li>\n<li class=\"github_commit\">                        <a href=\"https:\/\/github.com\/google\/trax\/commit\/7c189c2c3a37851df7fc641b322a99bbe9f8aa10\" target=\"_blank\">Configure tests to explicitly use jax_threefry_partitionable=False.See https:\/\/github.com\/jax-ml\/jax\/discussions\/18480PiperOrigin-RevId: 746173050<\/a>, Copybara-Service                    <\/li>\n<li class=\"github_commit\">                        <a href=\"https:\/\/github.com\/google\/trax\/commit\/3eb2e84c71acdfaa3a04a5454a4473b34d09b925\" target=\"_blank\">Automated Code ChangePiperOrigin-RevId: 724370877<\/a>, Copybara-Service                    <\/li>\n<li class=\"github_commit\">                        <a href=\"https:\/\/github.com\/google\/trax\/commit\/38fa93a2cd02f3bae55568df762e4a79bbd1c7e1\" target=\"_blank\">Avoid use of deprecated xla_bridge.get_backend().live_buffers()xla_bridge.get_backend is deprecated, and the public API for this is jax.live_arrays(). This is a drop-in replacement with no change of behavior.PiperOrigin-RevId: 723599353<\/a>, Copybara-Service                    <\/li>\n<\/ul>\n<\/details>\n<\/div>\n","protected":false},"excerpt":{"rendered":"","protected":false},"author":1,"featured_media":8519,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"footnotes":""},"categories":[9],"tags":[],"class_list":["post-8625","post","type-post","status-publish","format-standard","has-post-thumbnail","hentry","category-info"],"yoast_head":"<!-- This site is optimized with the Yoast SEO plugin v27.7 - https:\/\/yoast.com\/product\/yoast-seo-wordpress\/ -->\n<title>Trax: your path to advanced deep learning - Linux Shtuff<\/title>\n<meta name=\"robots\" content=\"index, follow, max-snippet:-1, max-image-preview:large, max-video-preview:-1\" \/>\n<link rel=\"canonical\" href=\"https:\/\/g33kinfo.com\/info\/trax-your-path-to-advanced-deep-learning\/\" \/>\n<meta property=\"og:locale\" content=\"en_US\" \/>\n<meta property=\"og:type\" content=\"article\" \/>\n<meta property=\"og:title\" content=\"Trax: your path to advanced deep learning - Linux Shtuff\" \/>\n<meta property=\"og:url\" content=\"https:\/\/g33kinfo.com\/info\/trax-your-path-to-advanced-deep-learning\/\" \/>\n<meta property=\"og:site_name\" content=\"Linux Shtuff\" \/>\n<meta property=\"article:publisher\" content=\"https:\/\/fb.me\/g33kinf0\" \/>\n<meta property=\"article:author\" content=\"https:\/\/fb.me\/g33kinf0\" \/>\n<meta property=\"article:published_time\" content=\"2020-02-17T16:36:45+00:00\" \/>\n<meta property=\"article:modified_time\" content=\"2020-02-17T16:36:47+00:00\" \/>\n<meta property=\"og:image\" content=\"https:\/\/g33kinfo.com\/info\/wp-content\/uploads\/2020\/01\/Minion_Research-e1587425764604.jpg\" \/>\n\t<meta property=\"og:image:width\" content=\"141\" \/>\n\t<meta property=\"og:image:height\" content=\"175\" \/>\n\t<meta property=\"og:image:type\" content=\"image\/jpeg\" \/>\n<meta name=\"author\" content=\"g33kadmin\" \/>\n<meta name=\"twitter:card\" content=\"summary_large_image\" \/>\n<meta name=\"twitter:creator\" content=\"@https:\/\/twitter.com\/drsinger1111\" \/>\n<meta name=\"twitter:site\" content=\"@drsinger1111\" \/>\n<script type=\"application\/ld+json\" class=\"yoast-schema-graph\">{\"@context\":\"https:\\\/\\\/schema.org\",\"@graph\":[{\"@type\":\"Article\",\"@id\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/trax-your-path-to-advanced-deep-learning\\\/#article\",\"isPartOf\":{\"@id\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/trax-your-path-to-advanced-deep-learning\\\/\"},\"author\":{\"name\":\"g33kadmin\",\"@id\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/#\\\/schema\\\/person\\\/c022e4c40b13ea1b678e6f020756f547\"},\"headline\":\"Trax: your path to advanced deep learning\",\"datePublished\":\"2020-02-17T16:36:45+00:00\",\"dateModified\":\"2020-02-17T16:36:47+00:00\",\"mainEntityOfPage\":{\"@id\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/trax-your-path-to-advanced-deep-learning\\\/\"},\"wordCount\":14,\"commentCount\":0,\"publisher\":{\"@id\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/#\\\/schema\\\/person\\\/c022e4c40b13ea1b678e6f020756f547\"},\"image\":{\"@id\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/trax-your-path-to-advanced-deep-learning\\\/#primaryimage\"},\"thumbnailUrl\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/wp-content\\\/uploads\\\/2020\\\/01\\\/Minion_Research-e1587425764604.jpg\",\"articleSection\":[\"General Info\"],\"inLanguage\":\"en-US\",\"potentialAction\":[{\"@type\":\"CommentAction\",\"name\":\"Comment\",\"target\":[\"https:\\\/\\\/g33kinfo.com\\\/info\\\/trax-your-path-to-advanced-deep-learning\\\/#respond\"]}]},{\"@type\":\"WebPage\",\"@id\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/trax-your-path-to-advanced-deep-learning\\\/\",\"url\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/trax-your-path-to-advanced-deep-learning\\\/\",\"name\":\"Trax: your path to advanced deep learning - Linux Shtuff\",\"isPartOf\":{\"@id\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/#website\"},\"primaryImageOfPage\":{\"@id\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/trax-your-path-to-advanced-deep-learning\\\/#primaryimage\"},\"image\":{\"@id\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/trax-your-path-to-advanced-deep-learning\\\/#primaryimage\"},\"thumbnailUrl\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/wp-content\\\/uploads\\\/2020\\\/01\\\/Minion_Research-e1587425764604.jpg\",\"datePublished\":\"2020-02-17T16:36:45+00:00\",\"dateModified\":\"2020-02-17T16:36:47+00:00\",\"breadcrumb\":{\"@id\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/trax-your-path-to-advanced-deep-learning\\\/#breadcrumb\"},\"inLanguage\":\"en-US\",\"potentialAction\":[{\"@type\":\"ReadAction\",\"target\":[\"https:\\\/\\\/g33kinfo.com\\\/info\\\/trax-your-path-to-advanced-deep-learning\\\/\"]}]},{\"@type\":\"ImageObject\",\"inLanguage\":\"en-US\",\"@id\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/trax-your-path-to-advanced-deep-learning\\\/#primaryimage\",\"url\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/wp-content\\\/uploads\\\/2020\\\/01\\\/Minion_Research-e1587425764604.jpg\",\"contentUrl\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/wp-content\\\/uploads\\\/2020\\\/01\\\/Minion_Research-e1587425764604.jpg\",\"width\":141,\"height\":175,\"caption\":\"Minion_Research\"},{\"@type\":\"BreadcrumbList\",\"@id\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/trax-your-path-to-advanced-deep-learning\\\/#breadcrumb\",\"itemListElement\":[{\"@type\":\"ListItem\",\"position\":1,\"name\":\"Home\",\"item\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/\"},{\"@type\":\"ListItem\",\"position\":2,\"name\":\"Trax: your path to advanced deep learning\"}]},{\"@type\":\"WebSite\",\"@id\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/#website\",\"url\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/\",\"name\":\"Linux Shtuff\",\"description\":\"Because I have CRS Syndrome...\",\"publisher\":{\"@id\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/#\\\/schema\\\/person\\\/c022e4c40b13ea1b678e6f020756f547\"},\"potentialAction\":[{\"@type\":\"SearchAction\",\"target\":{\"@type\":\"EntryPoint\",\"urlTemplate\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/?s={search_term_string}\"},\"query-input\":{\"@type\":\"PropertyValueSpecification\",\"valueRequired\":true,\"valueName\":\"search_term_string\"}}],\"inLanguage\":\"en-US\"},{\"@type\":[\"Person\",\"Organization\"],\"@id\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/#\\\/schema\\\/person\\\/c022e4c40b13ea1b678e6f020756f547\",\"name\":\"g33kadmin\",\"image\":{\"@type\":\"ImageObject\",\"inLanguage\":\"en-US\",\"@id\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/wp-content\\\/uploads\\\/2022\\\/07\\\/minion-researchA.gif\",\"url\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/wp-content\\\/uploads\\\/2022\\\/07\\\/minion-researchA.gif\",\"contentUrl\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/wp-content\\\/uploads\\\/2022\\\/07\\\/minion-researchA.gif\",\"width\":512,\"height\":512,\"caption\":\"g33kadmin\"},\"logo\":{\"@id\":\"https:\\\/\\\/g33kinfo.com\\\/info\\\/wp-content\\\/uploads\\\/2022\\\/07\\\/minion-researchA.gif\"},\"description\":\"I am a g33k, Linux blogger, developer, student and Tech Writer for Liquidweb.com\\\/kb. My passion for all things tech drives my hunt for all the coolz. I often need a vacation after I get back from vacation....\",\"sameAs\":[\"https:\\\/\\\/thelinuxreport.com\",\"https:\\\/\\\/fb.me\\\/g33kinf0\",\"https:\\\/\\\/x.com\\\/https:\\\/\\\/twitter.com\\\/drsinger1111\"]}]}<\/script>\n<!-- \/ Yoast SEO plugin. -->","yoast_head_json":{"title":"Trax: your path to advanced deep learning - Linux Shtuff","robots":{"index":"index","follow":"follow","max-snippet":"max-snippet:-1","max-image-preview":"max-image-preview:large","max-video-preview":"max-video-preview:-1"},"canonical":"https:\/\/g33kinfo.com\/info\/trax-your-path-to-advanced-deep-learning\/","og_locale":"en_US","og_type":"article","og_title":"Trax: your path to advanced deep learning - Linux Shtuff","og_url":"https:\/\/g33kinfo.com\/info\/trax-your-path-to-advanced-deep-learning\/","og_site_name":"Linux Shtuff","article_publisher":"https:\/\/fb.me\/g33kinf0","article_author":"https:\/\/fb.me\/g33kinf0","article_published_time":"2020-02-17T16:36:45+00:00","article_modified_time":"2020-02-17T16:36:47+00:00","og_image":[{"width":141,"height":175,"url":"https:\/\/g33kinfo.com\/info\/wp-content\/uploads\/2020\/01\/Minion_Research-e1587425764604.jpg","type":"image\/jpeg"}],"author":"g33kadmin","twitter_card":"summary_large_image","twitter_creator":"@https:\/\/twitter.com\/drsinger1111","twitter_site":"@drsinger1111","schema":{"@context":"https:\/\/schema.org","@graph":[{"@type":"Article","@id":"https:\/\/g33kinfo.com\/info\/trax-your-path-to-advanced-deep-learning\/#article","isPartOf":{"@id":"https:\/\/g33kinfo.com\/info\/trax-your-path-to-advanced-deep-learning\/"},"author":{"name":"g33kadmin","@id":"https:\/\/g33kinfo.com\/info\/#\/schema\/person\/c022e4c40b13ea1b678e6f020756f547"},"headline":"Trax: your path to advanced deep learning","datePublished":"2020-02-17T16:36:45+00:00","dateModified":"2020-02-17T16:36:47+00:00","mainEntityOfPage":{"@id":"https:\/\/g33kinfo.com\/info\/trax-your-path-to-advanced-deep-learning\/"},"wordCount":14,"commentCount":0,"publisher":{"@id":"https:\/\/g33kinfo.com\/info\/#\/schema\/person\/c022e4c40b13ea1b678e6f020756f547"},"image":{"@id":"https:\/\/g33kinfo.com\/info\/trax-your-path-to-advanced-deep-learning\/#primaryimage"},"thumbnailUrl":"https:\/\/g33kinfo.com\/info\/wp-content\/uploads\/2020\/01\/Minion_Research-e1587425764604.jpg","articleSection":["General Info"],"inLanguage":"en-US","potentialAction":[{"@type":"CommentAction","name":"Comment","target":["https:\/\/g33kinfo.com\/info\/trax-your-path-to-advanced-deep-learning\/#respond"]}]},{"@type":"WebPage","@id":"https:\/\/g33kinfo.com\/info\/trax-your-path-to-advanced-deep-learning\/","url":"https:\/\/g33kinfo.com\/info\/trax-your-path-to-advanced-deep-learning\/","name":"Trax: your path to advanced deep learning - Linux Shtuff","isPartOf":{"@id":"https:\/\/g33kinfo.com\/info\/#website"},"primaryImageOfPage":{"@id":"https:\/\/g33kinfo.com\/info\/trax-your-path-to-advanced-deep-learning\/#primaryimage"},"image":{"@id":"https:\/\/g33kinfo.com\/info\/trax-your-path-to-advanced-deep-learning\/#primaryimage"},"thumbnailUrl":"https:\/\/g33kinfo.com\/info\/wp-content\/uploads\/2020\/01\/Minion_Research-e1587425764604.jpg","datePublished":"2020-02-17T16:36:45+00:00","dateModified":"2020-02-17T16:36:47+00:00","breadcrumb":{"@id":"https:\/\/g33kinfo.com\/info\/trax-your-path-to-advanced-deep-learning\/#breadcrumb"},"inLanguage":"en-US","potentialAction":[{"@type":"ReadAction","target":["https:\/\/g33kinfo.com\/info\/trax-your-path-to-advanced-deep-learning\/"]}]},{"@type":"ImageObject","inLanguage":"en-US","@id":"https:\/\/g33kinfo.com\/info\/trax-your-path-to-advanced-deep-learning\/#primaryimage","url":"https:\/\/g33kinfo.com\/info\/wp-content\/uploads\/2020\/01\/Minion_Research-e1587425764604.jpg","contentUrl":"https:\/\/g33kinfo.com\/info\/wp-content\/uploads\/2020\/01\/Minion_Research-e1587425764604.jpg","width":141,"height":175,"caption":"Minion_Research"},{"@type":"BreadcrumbList","@id":"https:\/\/g33kinfo.com\/info\/trax-your-path-to-advanced-deep-learning\/#breadcrumb","itemListElement":[{"@type":"ListItem","position":1,"name":"Home","item":"https:\/\/g33kinfo.com\/info\/"},{"@type":"ListItem","position":2,"name":"Trax: your path to advanced deep learning"}]},{"@type":"WebSite","@id":"https:\/\/g33kinfo.com\/info\/#website","url":"https:\/\/g33kinfo.com\/info\/","name":"Linux Shtuff","description":"Because I have CRS Syndrome...","publisher":{"@id":"https:\/\/g33kinfo.com\/info\/#\/schema\/person\/c022e4c40b13ea1b678e6f020756f547"},"potentialAction":[{"@type":"SearchAction","target":{"@type":"EntryPoint","urlTemplate":"https:\/\/g33kinfo.com\/info\/?s={search_term_string}"},"query-input":{"@type":"PropertyValueSpecification","valueRequired":true,"valueName":"search_term_string"}}],"inLanguage":"en-US"},{"@type":["Person","Organization"],"@id":"https:\/\/g33kinfo.com\/info\/#\/schema\/person\/c022e4c40b13ea1b678e6f020756f547","name":"g33kadmin","image":{"@type":"ImageObject","inLanguage":"en-US","@id":"https:\/\/g33kinfo.com\/info\/wp-content\/uploads\/2022\/07\/minion-researchA.gif","url":"https:\/\/g33kinfo.com\/info\/wp-content\/uploads\/2022\/07\/minion-researchA.gif","contentUrl":"https:\/\/g33kinfo.com\/info\/wp-content\/uploads\/2022\/07\/minion-researchA.gif","width":512,"height":512,"caption":"g33kadmin"},"logo":{"@id":"https:\/\/g33kinfo.com\/info\/wp-content\/uploads\/2022\/07\/minion-researchA.gif"},"description":"I am a g33k, Linux blogger, developer, student and Tech Writer for Liquidweb.com\/kb. My passion for all things tech drives my hunt for all the coolz. I often need a vacation after I get back from vacation....","sameAs":["https:\/\/thelinuxreport.com","https:\/\/fb.me\/g33kinf0","https:\/\/x.com\/https:\/\/twitter.com\/drsinger1111"]}]}},"_links":{"self":[{"href":"https:\/\/g33kinfo.com\/info\/wp-json\/wp\/v2\/posts\/8625","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/g33kinfo.com\/info\/wp-json\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/g33kinfo.com\/info\/wp-json\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/g33kinfo.com\/info\/wp-json\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/g33kinfo.com\/info\/wp-json\/wp\/v2\/comments?post=8625"}],"version-history":[{"count":2,"href":"https:\/\/g33kinfo.com\/info\/wp-json\/wp\/v2\/posts\/8625\/revisions"}],"predecessor-version":[{"id":8627,"href":"https:\/\/g33kinfo.com\/info\/wp-json\/wp\/v2\/posts\/8625\/revisions\/8627"}],"wp:featuredmedia":[{"embeddable":true,"href":"https:\/\/g33kinfo.com\/info\/wp-json\/wp\/v2\/media\/8519"}],"wp:attachment":[{"href":"https:\/\/g33kinfo.com\/info\/wp-json\/wp\/v2\/media?parent=8625"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/g33kinfo.com\/info\/wp-json\/wp\/v2\/categories?post=8625"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/g33kinfo.com\/info\/wp-json\/wp\/v2\/tags?post=8625"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}