notebooks/03_pseudorandom_numbers_exercises.ipynb

{"metadata":{"kernelspec":{"language":"python","display_name":"Python 3","name":"python3"},"language_info":{"name":"python","version":"3.7.12","mimetype":"text/x-python","codemirror_mode":{"name":"ipython","version":3},"pygments_lexer":"ipython3","nbconvert_exporter":"python","file_extension":".py"}},"nbformat_minor":4,"nbformat":4,"cells":[{"cell_type":"markdown","source":"<center>\n    <h1>JaxTon</h1>\n    <i>💯 JAX exercises</i>\n    <br>\n    <br>\n    <a href='https://github.com/vopani/jaxton/blob/master/LICENSE'>\n        <img src='https://img.shields.io/badge/license-Apache%202.0-blue.svg?logo=apache'>\n    </a>\n    <a href='https://github.com/vopani/jaxton'>\n        <img src='https://img.shields.io/github/stars/vopani/jaxton?color=yellowgreen&logo=github'>\n    </a>\n    <a href='https://twitter.com/vopani'>\n        <img src='https://img.shields.io/twitter/follow/vopani'>\n    </a>\n</center>","metadata":{}},{"cell_type":"markdown","source":"<center>\n    This is Set 3: Pseudorandom Numbers (Exercises 21-30) of <b>JaxTon</b>: <i>💯 JAX exercises</i>\n    <br>\n    You can find all the exercises and solutions on <a href=\"https://github.com/vopani/jaxton#exercises-\">GitHub</a>\n</center>","metadata":{}},{"cell_type":"markdown","source":"**Prerequisites**\n\n* The configuration of jax should be set as shown in the code snippet below in order to use TPUs.\n* A sample array `sample_data` will be used for the exercises.","metadata":{}},{"cell_type":"code","source":"!python3 -m pip install jax","metadata":{"_kg_hide-input":true,"_kg_hide-output":true,"execution":{"iopub.status.busy":"2022-02-07T06:11:41.196161Z","iopub.execute_input":"2022-02-07T06:11:41.196526Z","iopub.status.idle":"2022-02-07T06:11:55.096828Z","shell.execute_reply.started":"2022-02-07T06:11:41.196435Z","shell.execute_reply":"2022-02-07T06:11:55.095242Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"## import packages\nimport jax\nimport jax.numpy as jnp\nimport os\nimport requests\n\n## setup JAX to use TPUs if available\ntry:\n    url = 'http:' + os.environ['TPU_NAME'].split(':')[1] + ':8475/requestversion/tpu_driver_nightly'\n    resp = requests.post(url)\n    jax.config.FLAGS.jax_xla_backend = 'tpu_driver'\n    jax.config.FLAGS.jax_backend_target = os.environ['TPU_NAME']\nexcept:\n    pass\n\njax.devices()","metadata":{"_kg_hide-input":false,"execution":{"iopub.status.busy":"2022-02-07T06:11:55.099712Z","iopub.execute_input":"2022-02-07T06:11:55.100136Z","iopub.status.idle":"2022-02-07T06:11:58.118425Z","shell.execute_reply.started":"2022-02-07T06:11:55.10008Z","shell.execute_reply":"2022-02-07T06:11:58.117686Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"code","source":"## sample data\nsample_data = jnp.array([10, 1, 24, 20, 15, 14])\nsample_data","metadata":{"execution":{"iopub.status.busy":"2022-02-07T06:11:58.119827Z","iopub.execute_input":"2022-02-07T06:11:58.120227Z","iopub.status.idle":"2022-02-07T06:11:58.138095Z","shell.execute_reply.started":"2022-02-07T06:11:58.120178Z","shell.execute_reply":"2022-02-07T06:11:58.13668Z"},"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"**Exercise 21: Create a pseudorandom number generator key with seed=100 and assign it to `key`**","metadata":{}},{"cell_type":"code","source":"","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"**Exercise 22: Create a subkey from `key` and assign it to `subkey`**","metadata":{}},{"cell_type":"code","source":"","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"**Exercise 23: Split `key` into seven subkeys `key_1`, `key_2`, `key_3`, `key_4`, `key_5`, `key_6` and `key_7`**","metadata":{}},{"cell_type":"code","source":"","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"**Exercise 24: Create a random permutation of `sample_data` using `key_1` and assign it to `data_permutation`**","metadata":{}},{"cell_type":"code","source":"","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"**Exercise 25: Choose a random element from `sample_data` using `key_2` and assign it to `random_selection`**","metadata":{}},{"cell_type":"code","source":"","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"**Exercise 26: Sample an integer between 10 and 24 using `key_3` and assign it to `sample_int`**","metadata":{}},{"cell_type":"code","source":"","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"**Exercise 27: Sample two values from uniform distribution between 1 and 2 using `key_4` and assign it to `sample_uniform`**","metadata":{}},{"cell_type":"code","source":"","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"**Exercise 28: Sample three values from bernoulli distribution using `key_5` and assign it to `sample_bernoulli`**","metadata":{}},{"cell_type":"code","source":"","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"**Exercise 29: Sample a 2x3 matrix from poisson distribution with λ=100 using `key_6` and assign it to `sample_poisson`**","metadata":{}},{"cell_type":"code","source":"","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"**Exercise 30: Sample a 2x3x4 array from normal distribution using `key_7` and assign it to `sample_normal`**","metadata":{}},{"cell_type":"code","source":"","metadata":{"trusted":true},"execution_count":null,"outputs":[]},{"cell_type":"markdown","source":"<center>\n    This completes Set 3: Pseudorandom Numbers (Exercises 21-30) of <b>JaxTon</b>: <i>💯 JAX exercises</i>\n    <br>\n    You can find all the exercises and solutions on <a href=\"https://github.com/vopani/jaxton#exercises-\">GitHub</a>\n</center>","metadata":{}}]}