This commit is contained in:
Morten Hjorth-Jensen
2023-05-03 11:48:50 -04:00
parent 356abdcde9
commit ceb79267af
3 changed files with 298 additions and 2 deletions
+18 -2
View File
@@ -4235,10 +4235,18 @@
},
{
"cell_type": "code",
"execution_count": 37,
"execution_count": 1,
"id": "e8fdc7be",
"metadata": {},
"outputs": [],
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"[0.25 0.19661197 0.10499357]\n"
]
}
],
"source": [
"import jax.numpy as jnp\n",
"from jax import grad, jit, vmap\n",
@@ -4250,6 +4258,14 @@
"derivative_fn = grad(sum_logistic)\n",
"print(derivative_fn(x_small))"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "1a930655",
"metadata": {},
"outputs": [],
"source": []
}
],
"metadata": {