diff --git a/Project.toml b/Project.toml
index ad88f2e..9eb4bd0 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,15 +1,17 @@
 name = "LuxCUDA"
 uuid = "d0bbae9a-e099-4d5b-a835-1c6931763bda"
 authors = ["Avik Pal <avikpal@mit.edu> and contributors"]
-version = "0.2.0"
+version = "0.2.1"
 
 [deps]
 CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
+NNlibCUDA = "a00861dc-f156-4864-bf3c-e6376f28a68d"
 Reexport = "189a3867-3050-52da-a836-e630ba90ab69"
 cuDNN = "02a925ec-e4fe-4b08-9a7e-0d78e3d38ccd"
 
 [compat]
 CUDA = "4"
+NNlibCUDA = "0.2"
 Reexport = "1"
 cuDNN = "1"
 julia = "1.6"
diff --git a/src/LuxCUDA.jl b/src/LuxCUDA.jl
index 766058d..c062082 100644
--- a/src/LuxCUDA.jl
+++ b/src/LuxCUDA.jl
@@ -2,7 +2,7 @@ module LuxCUDA
 
 using Reexport
 
-@reexport using CUDA, CUDA.CUDAKernels, cuDNN
+@reexport using CUDA, CUDA.CUDAKernels, cuDNN, NNlibCUDA
 
 const USE_CUDA_GPU = Ref{Union{Nothing, Bool}}(nothing)