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
• Incorrect gradients issue?
• ConvMixer
• MLPMixer ResNet
• Simple CNN

[codecov]

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 75.81%. Comparing base (3c3a432) to head (29a0cb9).

❗ There is a different number of reports uploaded between BASE (3c3a432) and HEAD (29a0cb9). Click for more details.

HEAD has 72 uploads less than BASE

Flag	BASE (3c3a432)	HEAD (29a0cb9)
	92	20

Additional details and impacted files

@@             Coverage Diff             @@
##             main    #1063       +/-   ##
===========================================
- Coverage   85.82%   75.81%   -10.01%     
===========================================
  Files         149      147        -2     
  Lines        6173     6145       -28     
===========================================
- Hits         5298     4659      -639     
- Misses        875     1486      +611     

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

[github-actions]

Benchmark Results (ASV)

	main	ec3c44e...	main/ec3c44e6c11390...
basics/overhead	0.123 ± 0.0011 μs	0.131 ± 0.0054 μs	0.939
time_to_load	0.9 ± 0.0041 s	0.902 ± 0.0029 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).

[avik-pal]

Not completely functional yet "error: expects input feature dimension (256) / feature_group_count = kernel input feature dimension (256). Got feature_group_count = 256."

xref: EnzymeAD/Reactant.jl#331

[avik-pal]

Unfortunately there is a strong discrepancy in the gradients between Zygote and Reactant. Need to debug