var (
	// FastNLPBert uses the fastNLP_Bert module, which is classified as a NLP Language Model.
	// fastNLP_Bert taxes the GPU heavily with low data movement. See Figure 2 on
	// page 5: https://arxiv.org/pdf/2304.14226.pdf
	//
	// https://github.com/pytorch/benchmark/tree/main/torchbenchmark/models/fastNLP_Bert
	// Bert Blog Post: https://towardsdatascience.com/bert-explained-state-of-the-art-language-model-for-nlp-f8b21a9b6270
	// Paper: https://arxiv.org/abs/1810.04805
	FastNLPBert = []pytorchTest{
		{
			// contains filtered or unexported fields
		},
		{
			// contains filtered or unexported fields
		},
	}

	// BigBird uses the hf_BigBird module, which is classified as a NLP Language Model.
	// hf_BigBird taxes the GPU moderately with low data movement. See Figure 2 on
	// page 5 (speech_tf): https://arxiv.org/pdf/2304.14226.pdf
	//
	// https://github.com/pytorch/benchmark/tree/main/torchbenchmark/models/hf_BigBird
	// Paper: https://arxiv.org/abs/2007.14062
	BigBird = []pytorchTest{
		{
			// contains filtered or unexported fields
		},
		{
			// contains filtered or unexported fields
		},
	}

	// SpeechTransformer uses the speech_transformer module classified as "Speech Recognition"
	// model. speech_transformer has a lot of idle time for the GPU. See Figure 2 on
	// page 5 (speech_tf): https://arxiv.org/pdf/2304.14226.pdf
	//
	// https://github.com/pytorch/benchmark/pull/374
	// Paper: https://arxiv.org/abs/1706.03762
	SpeechTransformer = []pytorchTest{
		{
			// contains filtered or unexported fields
		},
		{
			// contains filtered or unexported fields
		},
	}

	// LearningToPaint uses the LearningToPaint module classified as "neural renderer in model-based
	// Deep Reinforcement Learning (DRL)".
	// Learning to paint has a lot of "data movement" and doesn't tax the GPU a lot. See Figure 2 on
	// page 5: https://arxiv.org/pdf/2304.14226.pdf
	//
	// https://github.com/pytorch/benchmark/tree/main/torchbenchmark/models/LearningToPaint
	LearningToPaint = []pytorchTest{
		{
			// contains filtered or unexported fields
		},
		{
			// contains filtered or unexported fields
		},
	}

	// MobileNetV2 uses the mobilenet_v2 module classified as "Computer Vision: Image Classification".
	// MobileNet has a lot of taxes the GPU. See Figure 2 on page 5: https://arxiv.org/pdf/2304.14226.pdf
	//
	// https://github.com/pytorch/benchmark/tree/main/torchbenchmark/models/mobilenet_v2
	// Paper: https://paperswithcode.com/method/mobilenetv2
	MobileNetV2 = []pytorchTest{
		{
			// contains filtered or unexported fields
		},
		{
			// contains filtered or unexported fields
		},
	}

	// AllTests is a map of test names to the tests.
	AllTests = map[string][]pytorchTest{
		"FastNLPBert":       FastNLPBert,
		"BigBird":           BigBird,
		"SpeechTransformer": SpeechTransformer,
		"LearningToPaint":   LearningToPaint,
		"MobileNetV2":       MobileNetV2,
	}
)