pip install mamba
import mamba
python
import mamba
def compute(num):
return num * num
pool = mamba.Pool(processes=4)
results = pool.map(compute, [1, 2, 3, 4, 5])
print(results)
python
import mamba
import numpy as np
data = np.random.rand(1000000)
result = mamba.vectorize(np.sqrt)(data)
print(result)
python
import mamba
import numpy as np
data = np.random.rand(1000000)
result = mamba.to_gpu(data)
print(result)