From 99a9bcb73a689e73e6c20e6f163fd6c8f607ad3c Mon Sep 17 00:00:00 2001
From: Avik Pal <avikpal@mit.edu>
Date: Sun, 21 Jul 2024 18:33:19 -0700
Subject: [PATCH] chore: add NNlib as a direct dep

---
 Project.toml | 2 ++
 src/Lux.jl   | 1 +
 2 files changed, 3 insertions(+)

diff --git a/Project.toml b/Project.toml
index 1862ef2c1..125966d3e 100644
--- a/Project.toml
+++ b/Project.toml
@@ -24,6 +24,7 @@ LuxDeviceUtils = "34f89e08-e1d5-43b4-8944-0b49ac560553"
 LuxLib = "82251201-b29d-42c6-8e01-566dec8acb11"
 MacroTools = "1914dd2f-81c6-5fcd-8719-6d5c9610ff09"
 Markdown = "d6f4376e-aef5-505a-96c1-9c027394607a"
+NNlib = "872c559c-99b0-510c-b3b7-b6c96a88d5cd"
 Optimisers = "3bd65402-5787-11e9-1adc-39752487f4e2"
 Preferences = "21216c6a-2e73-6563-6e65-726566657250"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
@@ -98,6 +99,7 @@ MPI = "0.20.19"
 MacroTools = "0.5.13"
 Markdown = "1.10"
 NCCL = "0.1.1"
+NNlib = "0.9.21"
 OneHotArrays = "0.2.5"
 Optimisers = "0.3"
 Pkg = "1.10"
diff --git a/src/Lux.jl b/src/Lux.jl
index 30d83550a..c081fe9a1 100644
--- a/src/Lux.jl
+++ b/src/Lux.jl
@@ -18,6 +18,7 @@ using GPUArraysCore: GPUArraysCore, @allowscalar
 using LossFunctions: LossFunctions
 using MacroTools: MacroTools, block, combinedef, splitdef
 using Markdown: @doc_str
+using NNlib: NNlib
 using Optimisers: Optimisers
 using Preferences: load_preference, has_preference
 using Random: Random, AbstractRNG