torchbenchmark/models/pyhpc_isoneutral_mixing/__init__.py torchbenchmark/models/pyhpc_turbulent_kinetic_energy/tke_pytorch.py torchbenchmark/models/pyhpc_isoneutral_mixing/__init__.py torchbenchmark/models/pyhpc_turbulent_kinetic_energy/__init__.py torchbenchmark/models/pyhpc_isoneutral_mixing/isoneutral_pytorch.py torchbenchmark/models/pyhpc_turbulent_kinetic_energy/tke_pytorch.py torchbenchmark/models/pyhpc_isoneutral_mixing/isoneutral_pytorch.py torchbenchmark/models/pyhpc_turbulent_kinetic_energy/__init__.py torchbenchmark/models/pyhpc_isoneutral_mixing/metadata.yaml torchbenchmark/models/pyhpc_turbulent_kinetic_energy/metadata.yaml