Skip to content

Commit

Permalink
adjust directory for .ivy
Browse files Browse the repository at this point in the history
  • Loading branch information
Aarsh2001 committed Aug 9, 2023
1 parent 87ac913 commit 5f89622
Showing 1 changed file with 7 additions and 13 deletions.
20 changes: 7 additions & 13 deletions learn_the_basics/03_compile_code.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,10 @@
"metadata": {},
"outputs": [],
"source": [
"!pip install ivy"
"!pip install ivy\n",
"!pip install torch\n",
"!pip install jax\n",
"!pip intall jaxlib"
]
},
{
Expand All @@ -55,19 +58,10 @@
"def normalize(x):\n",
" mean = torch.mean(x)\n",
" std = torch.std(x)\n",
" return torch.div(torch.sub(x, mean), std)"
]
},
{
"cell_type": "code",
"execution_count": null,
"outputs": [],
"source": [
" return torch.div(torch.sub(x, mean), std)\n",
"\n",
"normalize = ivy.unify(normalize, source=\"torch\")"
],
"metadata": {
"collapsed": false
}
]
},
{
"attachments": {},
Expand Down

0 comments on commit 5f89622

Please sign in to comment.