Write down the Neural Tangents convolutional-layer inner loop as a Dex benchmark.
Write down the Neural Tangents convolutional-layer inner loop as a Dex benchmark.

google-research/dex-lang,google-research/dex-lang,google-research/dex-lang,google-research/dex-lang,google-research/dex-lang,google-research/dex-lang