|
14 | 14 | "id": "f4bb6871af102d56", |
15 | 15 | "metadata": { |
16 | 16 | "ExecuteTime": { |
17 | | - "end_time": "2024-07-17T21:26:11.700772Z", |
18 | | - "start_time": "2024-07-17T21:26:11.697782Z" |
| 17 | + "end_time": "2024-09-11T14:45:32.084217Z", |
| 18 | + "start_time": "2024-09-11T14:45:32.082Z" |
19 | 19 | } |
20 | 20 | }, |
21 | 21 | "outputs": [], |
|
37 | 37 | "id": "57fa7fa66ab3d5c0", |
38 | 38 | "metadata": { |
39 | 39 | "ExecuteTime": { |
40 | | - "end_time": "2024-07-17T21:26:12.634248Z", |
41 | | - "start_time": "2024-07-17T21:26:11.701790Z" |
| 40 | + "end_time": "2024-09-11T14:45:32.090478Z", |
| 41 | + "start_time": "2024-09-11T14:45:32.088273Z" |
42 | 42 | } |
43 | 43 | }, |
44 | 44 | "outputs": [], |
45 | 45 | "source": [ |
46 | | - "import os\n", |
47 | | - "from functools import partial\n", |
48 | | - "\n", |
49 | 46 | "import jax\n", |
50 | 47 | "import jax.numpy as jnp\n", |
51 | 48 | "import matplotlib.pyplot as plt\n", |
52 | 49 | "import optax\n", |
53 | | - "from numpyro.infer import MCMC, NUTS\n", |
54 | 50 | "\n", |
55 | 51 | "from healpix_geometry_analysis.coordinates import HealpixCoordinates\n", |
56 | 52 | "from healpix_geometry_analysis.geometry.tile import TileGeometry\n", |
57 | | - "from healpix_geometry_analysis.problems.numpyro_sampler import NumpyroSamplerProblem\n", |
58 | 53 | "from healpix_geometry_analysis.problems.optax_optimizer import OptaxOptimizerProblem\n", |
59 | 54 | "from healpix_geometry_analysis.enable_x64 import enable_x64\n", |
60 | 55 | "\n", |
| 56 | + "# Use float64 with Jax\n", |
61 | 57 | "enable_x64()" |
62 | 58 | ] |
63 | 59 | }, |
|
75 | 71 | "id": "3bc13fcf4eb9c747", |
76 | 72 | "metadata": { |
77 | 73 | "ExecuteTime": { |
78 | | - "end_time": "2024-07-17T21:26:12.655671Z", |
79 | | - "start_time": "2024-07-17T21:26:12.634963Z" |
| 74 | + "end_time": "2024-09-11T14:45:32.098947Z", |
| 75 | + "start_time": "2024-09-11T14:45:32.097078Z" |
80 | 76 | } |
81 | 77 | }, |
82 | 78 | "outputs": [], |
|
98 | 94 | "id": "69b3011e7a38f3b9", |
99 | 95 | "metadata": { |
100 | 96 | "ExecuteTime": { |
101 | | - "end_time": "2024-07-17T21:26:13.074963Z", |
102 | | - "start_time": "2024-07-17T21:26:12.656748Z" |
| 97 | + "end_time": "2024-09-11T14:45:32.108669Z", |
| 98 | + "start_time": "2024-09-11T14:45:32.104440Z" |
103 | 99 | } |
104 | 100 | }, |
105 | 101 | "outputs": [], |
|
114 | 110 | "id": "c859164b16234847", |
115 | 111 | "metadata": { |
116 | 112 | "ExecuteTime": { |
117 | | - "end_time": "2024-07-17T21:26:13.414059Z", |
118 | | - "start_time": "2024-07-17T21:26:13.075581Z" |
| 113 | + "end_time": "2024-09-11T14:45:32.302155Z", |
| 114 | + "start_time": "2024-09-11T14:45:32.187751Z" |
119 | 115 | } |
120 | 116 | }, |
121 | 117 | "outputs": [], |
|
143 | 139 | "id": "22d4a055b369364c", |
144 | 140 | "metadata": { |
145 | 141 | "ExecuteTime": { |
146 | | - "end_time": "2024-07-17T21:26:13.416602Z", |
147 | | - "start_time": "2024-07-17T21:26:13.414815Z" |
| 142 | + "end_time": "2024-09-11T14:45:32.310509Z", |
| 143 | + "start_time": "2024-09-11T14:45:32.308439Z" |
148 | 144 | } |
149 | 145 | }, |
150 | 146 | "outputs": [], |
|
197 | 193 | "id": "8a340519783e6a84", |
198 | 194 | "metadata": { |
199 | 195 | "ExecuteTime": { |
200 | | - "end_time": "2024-07-17T21:26:21.587926Z", |
201 | | - "start_time": "2024-07-17T21:26:13.417116Z" |
| 196 | + "end_time": "2024-09-11T14:45:40.133132Z", |
| 197 | + "start_time": "2024-09-11T14:45:32.316469Z" |
202 | 198 | } |
203 | 199 | }, |
204 | 200 | "outputs": [], |
|
0 commit comments