Endpoints — github.com/Qualcomm-AI-research/geometric-algebra-transformer

CodeHub MCPcopy Create free account

hub / github.com/Qualcomm-AI-research/geometric-algebra-transformer / endpoints

Endpoints4 in github.com/Qualcomm-AI-research/geometric-algebra-transformer

⨍ Functions 452◇ Types & classes 61 ↳ Endpoints 4

Route_lin_square_normalizer

minimum_autocast_precision(torch.float32, output="low")

gatr/primitives/attention.py:None

minimum_autocast_precision(torch.float32, output=output_mode)

tests/gatr/utils/test_misc.py:None

Routetest_linear_equivariance

pytest.mark.parametrize( "input_batch_dims,coeff_batch_dims", [ ((7,), (5, 7)), ((3, 7

tests/gatr/primitives/test_linear.py:None

Routetest_minimum_autocast_precision_outputs

pytest.mark.parametrize( "output_mode,expected_dtype", [ (None, torch.float64), (torch

tests/gatr/utils/test_misc.py:None