SciMLBenchmarks.jl
SciMLBenchmarks.jl copied to clipboard
[WIP] perf: common neural network workloads benchmarks
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
- [ ] Simple MLP
- [ ] Add Reactant Inference Code
- Training code will have to wait till the
compile
PR lands in Lux
- Training code will have to wait till the