diff --git a/REQUIRE b/REQUIRE index b75b59ca..8bb92ddb 100644 --- a/REQUIRE +++ b/REQUIRE @@ -9,6 +9,7 @@ GZip Colors ZipFile AbstractTrees +Reexport # AD ForwardDiff 0.5.0 diff --git a/src/Flux.jl b/src/Flux.jl index 939d6e2c..c1e18115 100644 --- a/src/Flux.jl +++ b/src/Flux.jl @@ -4,7 +4,7 @@ module Flux # Zero Flux Given -using Juno, Requires +using Juno, Requires, Reexport using MacroTools: @forward export Chain, Dense, RNN, LSTM, GRU, Conv2D, @@ -12,9 +12,7 @@ export Chain, Dense, RNN, LSTM, GRU, Conv2D, SGD, ADAM, Momentum, Nesterov, AMSGrad, param, params, mapleaves -using NNlib -export σ, sigmoid, logσ, logsigmoid, relu, leakyrelu, elu, swish, softmax, logsoftmax, - conv2d, maxpool2d, avgpool2d +@reexport using NNlib include("tracker/Tracker.jl") using .Tracker