Skip to content
This repository has been archived by the owner on Jun 14, 2024. It is now read-only.

Switch to deep numpy as backend. #200

Open
xidulu opened this issue Sep 5, 2019 · 0 comments
Open

Switch to deep numpy as backend. #200

xidulu opened this issue Sep 5, 2019 · 0 comments

Comments

@xidulu
Copy link

xidulu commented Sep 5, 2019

Hi,

I'm a member of Deep Numpy (a new MXNet frontend API with numpy like interface: https://numpy.mxnet.io/index.html) dev team and I'm currently working on the random module for Deep Numpy, which, as its name, has behavior exactly like Numpy, including broadcastable parameters and output shape, for example:
apache/mxnet#15858

Also, sampling backend for rejection sampling is entirely rewritten to remove while loop from GPU Kernel. The new version, according to my profiling, performs ten times faster than nd.random on GPU at the cost of tiny increase in memory usage. (not merged yet)
apache/mxnet#15928

We could have some further discussion if you have interests in switching to deep numpy's random sampling backend, or, migrating MXFusion to Deep Numpy API :-)

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant