Skip to content

Instantly share code, notes, and snippets.

View modanesh's full-sized avatar
:bowtie:

Mohamad H. Danesh modanesh

:bowtie:
View GitHub Profile
@modanesh
modanesh / weird_jax_behavior.py
Last active October 17, 2022 14:10
Value of `float(img.mean())` differs before and after `jax_disable_jit`
import cv2
import imageio
import jax
import os
import urllib
import numpy as np
import jax.numpy as jnp
def resize_and_center_crop(image):
@modanesh
modanesh / rejection_sampling.py
Last active December 10, 2021 04:29
Generate normally-distributed random samples from uniform samples by rejection sampling
import numpy as np
import matplotlib.pyplot as plt
import bisect
from scipy import stats
rng = np.random.default_rng()
desired_variance = 1
desired_mean = 0
uni_rand = rng.uniform(low=-desired_variance*3, high=desired_variance*3, size=1000)
_ = plt.hist(uni_rand, bins='auto')
@modanesh
modanesh / box_muller.py
Created December 9, 2021 07:07
Box-Muller transformation
import math
import random
import matplotlib.pyplot as plt
uni_rand_1 = []
uni_rand_2 = []
gauss_samples_1 = []
gauss_samples_2 = []
for i in range(1000):
random_1 = random.random()