tracel-ai/burn v0.5.0

on GitHub

New Modules for Vision Tasks

• Conv1D, Conv2D currently without support for stride, dilation, or group convolution
• MaxPool2D
• BatchNorm2D

New General Tensor Operations

• log1p thanks to @bioinformatist
• sin, cos, tanh thanks to @makroiss

Breaking Changes

• Devices are now passed by reference, thanks to feedback from @djdisodo.
• The shape function now returns an owned struct, and backends no longer need to cache each shape.