SciMLBenchmarks.jl icon indicating copy to clipboard operation
SciMLBenchmarks.jl copied to clipboard

[WIP] perf: common neural network workloads benchmarks

Open avik-pal opened this issue 6 months ago • 3 comments

currently add CPU versions. should be easy to extend to GPU once SciMLBenchmarks has GPU runners available.

TODOs

  • [ ] Restore Manifest to use the released versions of packages
  • [ ] Boilerplate
    • [x] Inference
    • [x] Inference Plots
    • [ ] Training
    • [ ] Training Plots
  • [ ] Models
    • [ ] Simple MLP
      • [x] Inference
      • [ ] Training
    • [ ] Simple MLP with BN -- wait on https://github.com/LuxDL/LuxLib.jl/pull/126
      • [ ] Inference
      • [ ] Training
    • [ ] LeNet
      • [x] Inference
      • [ ] Training
    • [ ] ResNet
      • [ ] Inference
      • [ ] Training
  • [ ] Add Reactant Inference Code
    • Training code will have to wait till the compile PR lands in Lux

avik-pal avatar Aug 11 '24 21:08 avik-pal