feat: update ConvMixer to support reactant

[avik-pal]

• Needs conv adjoints conv gradient is not implemented in EnzymeJAX EnzymeAD/Reactant.jl#214
• Needs reduce_window adjoint -- we could potentially work around this

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 37.18%. Comparing base (ed0d75c) to head (090b87e).

❗ There is a different number of reports uploaded between BASE (ed0d75c) and HEAD (090b87e). Click for more details.

HEAD has 55 uploads less than BASE

Flag	BASE (ed0d75c)	HEAD (090b87e)
	56	1

Additional details and impacted files

@@             Coverage Diff             @@
##             main    #1063       +/-   ##
===========================================
- Coverage   84.98%   37.18%   -47.80%     
===========================================
  Files         145       60       -85     
  Lines        6032     2896     -3136     
===========================================
- Hits         5126     1077     -4049     
- Misses        906     1819      +913     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[github-actions]

Benchmark Results (ASV)

	main	ee2fb4f...	main/ee2fb4f4803cc9...
basics/overhead	0.121 ± 0.0011 μs	0.124 ± 0.0011 μs	0.978
time_to_load	1.19 ± 0.0083 s	1.2 ± 0.016 s	0.998

Benchmark Plots

A plot of the benchmark results have been uploaded as an artifact to the workflow run for this PR.
Go to "Actions"->"Benchmark a pull request"->[the most recent run]->"Artifacts" (at the bottom).