Classifying Galaxy Images with a neural-network written in PyTorch
An example of how you can write a convolutional neural-network in PyTorch to classify 2D galaxy images. There is a pdf describing this exercise and associated code to replicate the results
