diff --git a/.buildkite/pipeline.yml b/.buildkite/pipeline.yml
index dafb761..acc3050 100644
--- a/.buildkite/pipeline.yml
+++ b/.buildkite/pipeline.yml
@@ -19,9 +19,11 @@ steps:
         julia:
           - "1"
           - "1.6"
-          - "1.9-nightly"
           - "nightly"
       adjustments:
+        - with:
+            julia: "1.6"
+          soft_fail: true
         - with:
             julia: "nightly"
           soft_fail: true
diff --git a/.github/workflows/CI.yml b/.github/workflows/CI.yml
index 697a2bd..cab3a0e 100644
--- a/.github/workflows/CI.yml
+++ b/.github/workflows/CI.yml
@@ -20,7 +20,6 @@ jobs:
         version:
           - "1"
           - "1.6"
-          - "~1.9.0-0"
     steps:
       - uses: actions/checkout@v3
       - uses: julia-actions/setup-julia@v1
@@ -44,4 +43,3 @@ jobs:
       - uses: codecov/codecov-action@v3
         with:
           files: lcov.info
-          flags: ${{ matrix.group }}
diff --git a/Project.toml b/Project.toml
index 34d58c4..ad88f2e 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,17 +1,15 @@
 name = "LuxCUDA"
 uuid = "d0bbae9a-e099-4d5b-a835-1c6931763bda"
 authors = ["Avik Pal <avikpal@mit.edu> and contributors"]
-version = "0.1.2"
+version = "0.2.0"
 
 [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.1"
-NNlibCUDA = "0.2"
+CUDA = "4"
 Reexport = "1"
 cuDNN = "1"
 julia = "1.6"
diff --git a/src/LuxCUDA.jl b/src/LuxCUDA.jl
index 4de5070..766058d 100644
--- a/src/LuxCUDA.jl
+++ b/src/LuxCUDA.jl
@@ -2,7 +2,7 @@ module LuxCUDA
 
 using Reexport
 
-@reexport using CUDA, CUDA.CUDAKernels, NNlibCUDA, cuDNN
+@reexport using CUDA, CUDA.CUDAKernels, cuDNN
 
 const USE_CUDA_GPU = Ref{Union{Nothing, Bool}}(nothing)