Benchmark for Transformers

FluxML / FluxBench.jl

Benchmarks for the FluxML ecosystem for deep learning, scientific machine learning, differentiable programming etc including AD and CUDA accelerated workloads

https://speed.fluxml.ai

14 stars 9 forks source link

Benchmark for Transformers #7

Closed chengchingwen closed 3 years ago

DhairyaLGandhi commented 3 years ago

Thank you for this! This looks good to me, we would need CI to run on it.

DhairyaLGandhi commented 3 years ago

Would you mind if I convert it from a draft to a regular PR so CI can be run?

chengchingwen commented 3 years ago

I'm adding another benchmark and making sure it doesn't exceed the memory limit. I'll convert to a regular once I finish it.

DhairyaLGandhi commented 3 years ago

lgtm, let me know when you're happy and we can merge this

chengchingwen commented 3 years ago

I think it's ready to go

DhairyaLGandhi commented 3 years ago

Cool, thanks @chengchingwen !