Add reactant forward and reverse pass tests (#2576)

Author: wsmoses

.buildkite/pipeline.yml

@@ -15,6 +15,7 @@ steps:
       FLUX_TEST_CUDA: "true"
       FLUX_TEST_CPU: "false"
       FLUX_TEST_ENZYME: "false"
+      FLUX_TEST_REACTANT: "true"
     timeout_in_minutes: 60
 
   - label: "Metal - Julia 1"
@@ -37,6 +38,7 @@ steps:
       FLUX_TEST_METAL: "true"
       FLUX_TEST_CPU: "false"
       FLUX_TEST_ENZYME: "false"
+      FLUX_TEST_REACTANT: "true"
 
   - label: "AMDGPU - Julia 1"
     plugins:
@@ -59,6 +61,7 @@ steps:
       FLUX_TEST_AMDGPU: "true"
       FLUX_TEST_CPU: "false"
       FLUX_TEST_ENZYME: "false"
+      FLUX_TEST_REACTANT: "false"
       JULIA_NUM_THREADS: 4
 
 env:

Project.toml

@@ -18,6 +18,7 @@ Optimisers = "3bd65402-5787-11e9-1adc-39752487f4e2"
 Preferences = "21216c6a-2e73-6563-6e65-726566657250"
 ProgressLogging = "33c8b6b6-d38a-422a-b730-caa89a2f386c"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
+Reactant = "3c362404-f566-11ee-1572-e11a4b42c853"
 Reexport = "189a3867-3050-52da-a836-e630ba90ab69"
 Setfield = "efcf1570-3423-57d1-acb7-fd33fddbac46"
 SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
@@ -60,6 +61,7 @@ OneHotArrays = "0.2.4"
 Optimisers = "0.4.1"
 Preferences = "1"
 ProgressLogging = "0.1"
+Reactant = "0.2.16"
 Reexport = "1.0"
 Setfield = "1.1"
 SpecialFunctions = "2.1.2"

test/Project.toml

@@ -14,6 +14,7 @@ MLUtils = "f1d291b0-491e-4a28-83b9-f70985020b54"
 Optimisers = "3bd65402-5787-11e9-1adc-39752487f4e2"
 Pkg = "44cfe95a-1eb2-52ea-b672-e2afdf69b78f"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
+Reactant = "3c362404-f566-11ee-1572-e11a4b42c853"
 SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
 Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"