1041: add NNlib docs + misc docs improvements r=CarloLucibello a=CarloLucibello
Partially addressing https://github.com/FluxML/NNlib.jl/issues/137.
Also, I'm leaving out the `σ` activation and using its alias `sigmoid`, since `σ` conveys little information and it is also used to denote a generic activation in the Dense layer. I think we should deprecate `σ` in NNlib, has this been discussed already?
In an ideal world, before merging this, we should get NNlib to either unexport or add docs to its undocumented exports
Co-authored-by: Carlo Lucibello <carlo.lucibello@gmail.com>
998: test restructure on the GPU r=CarloLucibello a=ChrisRackauckas
Requires https://github.com/FluxML/Zygote.jl/pull/474 to pass
Co-authored-by: Chris Rackauckas <accounts@chrisrackauckas.com>
960: Added utility function outdims to compute output dimensions of a layer r=dhairyagandhi96 a=darsnack
Based on Slack chatter, I added a utility function, `outdims`, that computes the output dimensions for given input dimensions.
Example
```julia
layer = Conv((3, 3), 3 => 16)
outdims(layer, (10, 10)) # returns (8, 8)
```
Co-authored-by: Kyle Daruwalla <daruwalla@wisc.edu>
1013: Adapt to GPUArrays/CuArrays changes r=dhairyagandhi96 a=maleadt
Changes in response to a29df67184 and https://github.com/JuliaGPU/CuArrays.jl/pull/576. I suppose the next CuArrays release will need to be breaking because of this.
Maybe the `crossentropy` signature needs to be adjusted to support integer vectors, but I'll leave that decision up to Flux developers. This at least is the quick fix to get the tests passing again.
Co-authored-by: Tim Besard <tim.besard@gmail.com>
680: Added new loss functions. r=thebhatman a=thebhatman
I have added the KL Divergence Loss function, Poisson loss function, Logcosh loss, and Hinge loss function.
Co-authored-by: Manjunath Bhat <manjunathbhat9920@gmail.com>
Co-authored-by: thebhatman <manjunathbhat9920@gmail.com>
992: Compat bounds for a couple more packages r=dhairyagandhi96 a=dhairyagandhi96
adds compatibility bounds for a few more packages
cc @MikeInnes
Co-authored-by: Dhairya Gandhi <dhairya@juliacopmuting.com>
bors[bot]