Skip to content

Instantly share code, notes, and snippets.

View darkmatter18's full-sized avatar
:electron:
In relationship with console.log()

Arkadip Bhattacharya darkmatter18

:electron:
In relationship with console.log()
View GitHub Profile

PyTorch to MXNet

This cheatsheet serves as a quick reference for PyTorch users who are interested in trying MXNet, and vice versa.

Pytorch is a deep learning framework provides imperative tensor manipulation and neural network training. MXNet provides similar imperative tensor manipulation through the ndarray package and neural network training through gluon. This cheatsheet maps functions one-by-one between these two frameworks.

Note that MXNet has a symbolic interface similar to Keras and Tensorflow that may provide better performance and portability. This cheatsheet mainly focus on MXNet's imperative interface.

Installation