Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Qualcomm-AI-research/geometric-algebra-transformer
/ functions
Functions
452 in github.com/Qualcomm-AI-research/geometric-algebra-transformer
⨍
Functions
452
◇
Types & classes
61
↳
Endpoints
4
↓ 1 callers
Function
sample_uniform_in_circle
Samples uniformly in a 2D circle using batched rejection sampling.
gatr/utils/misc.py:75
↓ 1 callers
Function
sandwich
Given clifford multivectors, computes their sandwich product. Specifically, given a Pin element u and a PGA element x, both given as clifford mul
gatr/utils/clifford.py:85
↓ 1 callers
Function
sdp_attention
Equivariant geometric attention based on scaled dot products. Expects both multivector and scalar queries, keys, and values as inputs. Then t
gatr/primitives/attention.py:38
↓ 1 callers
Method
should_refactor
Decides whether a file object should be refactored or not.
docker/ext_packages/install_upstream_python_packages.py:197
↓ 1 callers
Function
sum_
(*args)
tests/gatr/utils/test_misc.py:78
↓ 1 callers
Function
tensor_to_mv
Shorthand to transform a numpy array to a PGA multivector.
gatr/utils/clifford.py:18
↓ 1 callers
Function
verify_installation_through_imports
Tests if installation was successful by importing the installed modules.
docker/ext_packages/install_upstream_python_packages.py:270
↓ 1 callers
Method
visualize
Visualization function. To be implemented by subclasses.
gatr/experiments/base_experiment.py:759
Method
__call__
Apply Pin transformation to multivector inputs.
gatr/utils/clifford.py:115
Method
__call__
Allows to access cached function values with `()` instead of `[]`.
gatr/utils/einsum.py:65
Method
__call__
Performs experiment as outlined below. - initializes all the logistics - instantiates model (if necessary) - loads checkpoint
gatr/experiments/base_experiment.py:76
Method
__getitem__
(self, item)
tests_regression/regression_helpers.py:15
Method
__getitem__
Return datapoint.
tests_regression/regression_datasets/translation_norm.py:29
Method
__getitem__
Return datapoint.
tests_regression/regression_datasets/translate_point.py:31
Method
__getitem__
Return datapoint.
tests_regression/regression_datasets/points_distance.py:31
Method
__getitem__
Return datapoint.
tests_regression/regression_datasets/connect_points.py:29
Method
__getitem__
Return an item from the dataset.
gatr/experiments/arteries/dataset.py:53
Method
__getitem__
Returns the `idx`-th sample from the dataset.
gatr/experiments/nbody/dataset.py:34
Method
__init__
(self, dataset)
tests_regression/regression_helpers.py:12
Method
__init__
(self, model)
tests_regression/regression_helpers.py:26
Method
__init__
(self, net)
tests_regression/regression_datasets/translation_norm.py:44
Method
__init__
(self, net)
tests_regression/regression_datasets/translate_point.py:46
Method
__init__
(self, net)
tests_regression/regression_datasets/points_distance.py:46
Method
__init__
(self, net)
tests_regression/regression_datasets/connect_points.py:44
Method
__init__
( self, in_channels, out_channels, message_channels, mlp_hidden_channe
gatr/baselines/gcan.py:192
Method
__init__
( self, in_channels, out_channels, node_channels, message_channels,
gatr/baselines/gcan.py:264
Method
__init__
(self, in_shape, out_shape, hidden_channels, hidden_layers)
gatr/baselines/mlp.py:16
Method
__init__
(self, in_channels, hidden_channels, num_heads)
gatr/baselines/transformer.py:94
Method
__init__
( self, in_channels: int, out_channels: int, hidden_channels: int, num
gatr/baselines/transformer.py:150
Method
__init__
( self, channels, num_heads: int = 8, pos_encoding: bool = False, pos_
gatr/baselines/transformer.py:261
Method
__init__
( self, in_channels: int, out_channels: int, hidden_channels: int, num
gatr/baselines/transformer.py:356
Method
__init__
( self, in_channels: int, out_channels: int, hidden_channels: int, num
gatr/baselines/transformer.py:448
Method
__init__
( self, input_vec_channels=2, input_s_channels=1, output_vec_channels=1,
gatr/baselines/segnn.py:61
Method
__init__
(self, spin=False, rng=None)
gatr/utils/clifford.py:110
Method
__init__
(self, fn: Callable)
gatr/utils/einsum.py:55
Method
__init__
( self, in_mv_channels: int, out_mv_channels: int, in_s_channels: Optional[int
gatr/utils/compile_linear.py:34
Method
__init__
( self, in_mv_channels: int, out_mv_channels: int, hidden_mv_channels: int,
gatr/nets/gatr.py:62
Method
__init__
( self, in_mv_channels: int, out_mv_channels: int, hidden_mv_channels: int,
gatr/nets/axial_gatr.py:68
Method
__init__
(self, cfg)
gatr/experiments/base_experiment.py:51
Method
__init__
(self, net, scalars=True, return_other=True)
gatr/experiments/base_wrapper.py:23
Method
__init__
(self, num, special, seed)
gatr/experiments/arteries/dataset.py:38
Method
__init__
(self, cfg)
gatr/experiments/arteries/experiment.py:35
Method
__init__
(self, net)
gatr/experiments/arteries/wrappers.py:42
Method
__init__
( self, time=0.1, time_steps=100, star_mass_range=(1.0, 10.0), planet_
gatr/experiments/nbody/simulator.py:53
Method
__init__
(self, filename, subsample=None, keep_trajectories=False)
gatr/experiments/nbody/dataset.py:24
Method
__init__
(self, cfg)
gatr/experiments/nbody/experiment.py:20
Method
__init__
(self, net)
gatr/experiments/nbody/wrappers.py:146
Method
__init__
(self, net, neighbors, lmax_attr, canonicalize_mode="com")
gatr/experiments/nbody/wrappers.py:182
Method
__init__
(self, net, canonicalize_to_com=True, canonicalize_mode="com")
gatr/experiments/nbody/wrappers.py:297
Method
__init__
(self, net, geometric_batching=False)
gatr/experiments/nbody/wrappers.py:410
Method
__init__
( self, in_mv_channels: int, out_mv_channels: int, in_s_channels: Optional[int
gatr/layers/linear.py:64
Method
__init__
(self, p: float = 0.0)
gatr/layers/dropout.py:22
Method
__init__
( self, mv_channels: int, s_channels: int, attention: SelfAttentionConfig,
gatr/layers/gatr_block.py:43
Method
__init__
(self, mv_channel_dim=-2, scalar_channel_dim=-1, epsilon: float = 0.01)
gatr/layers/layer_norm.py:33
Method
__init__
(self, num_channels, item_dim, base=4096)
gatr/layers/attention/positional_encoding.py:45
Method
__init__
( self, config: SelfAttentionConfig, in_q_mv_channels: int, in_q_s_channels: O
gatr/layers/attention/cross_attention.py:32
Method
__init__
(self, config: SelfAttentionConfig)
gatr/layers/attention/qkv.py:109
Method
__init__
(self, config: SelfAttentionConfig)
gatr/layers/attention/self_attention.py:30
Method
__init__
(self, config: SelfAttentionConfig)
gatr/layers/attention/attention.py:38
Method
__init__
(self, nonlinearity: str = "relu", **kwargs)
gatr/layers/mlp/nonlinearities.py:24
Method
__init__
( self, config: MLPConfig, )
gatr/layers/mlp/mlp.py:33
Method
__init__
( self, in_mv_channels: int, out_mv_channels: int, hidden_mv_channels: Optiona
gatr/layers/mlp/geometric_bilinears.py:34
Method
__init__
( self, name: str, clone_url: str, checksum_sha: str, checkout_base_di
docker/ext_packages/install_upstream_python_packages.py:57
Method
__len__
(self)
tests_regression/regression_helpers.py:19
Method
__len__
Return number of samples.
tests_regression/regression_datasets/translation_norm.py:25
Method
__len__
Return number of samples.
tests_regression/regression_datasets/translate_point.py:27
Method
__len__
Return number of samples.
tests_regression/regression_datasets/points_distance.py:27
Method
__len__
Return number of samples.
tests_regression/regression_datasets/connect_points.py:25
Method
__len__
Dataset size.
gatr/experiments/arteries/dataset.py:49
Method
__len__
Returns the number of samples in the dataset.
gatr/experiments/nbody/dataset.py:30
Method
__missing__
Computes missing function values and adds them to the cache.
gatr/utils/einsum.py:59
Method
__post_init__
Type checking / conversion.
gatr/layers/attention/config.py:66
Method
__post_init__
Type checking / conversion.
gatr/layers/mlp/config.py:31
Method
_compute_metrics
Given a dataloader, computes all relevant metrics. To be implemented by subclasses. Parameters ---------- dataloader : torch.
gatr/experiments/arteries/experiment.py:167
Method
_construct_join_reference
Constructs a reference vector for dualization from the inputs.
gatr/nets/axial_gatr.py:194
Function
_einsum_with_path_ignored
Calls torch.einsum whilst dropping all kwargs. Allows use of hard-coded optimal contraction paths in `gatr_einsum_with_path` for non-compilin
gatr/utils/einsum.py:21
Method
_eval_dataset_tags
Eval dataset tags, to be implemented by subclasses. Returns ------- tags : iterable of str Eval dataset tags
gatr/experiments/base_experiment.py:766
Method
_eval_dataset_tags
Eval dataset tags. Returns ------- tags : iterable of str Eval dataset tags
gatr/experiments/arteries/experiment.py:230
Method
_eval_dataset_tags
Eval dataset tags. Returns ------- tags : iterable of str Eval dataset tags
gatr/experiments/nbody/experiment.py:85
Method
_forward
Model forward pass. Parameters ---------- data : tuple of torch.Tensor Data batch. Returns -----
gatr/experiments/arteries/experiment.py:129
Method
_forward
Model forward pass. Parameters ---------- data : tuple of torch.Tensor Data batch. Returns -----
gatr/experiments/nbody/experiment.py:50
Method
_get_refactored_path
(self)
docker/ext_packages/install_upstream_python_packages.py:94
Function
_lin_square_normalizer
Apply linear square normalization to the input tensor. Parameters ---------- v : Tensor Input tensor. epsilon : float, option
gatr/primitives/attention.py:265
Method
_load_dataset
Loads dataset. References ---------- J. Suk et al., "Mesh Convolutional Neural Networks for Wall Shear Stress Estimation in
gatr/experiments/arteries/experiment.py:56
Method
_load_dataset
Loads dataset. Parameters ---------- tag : str Dataset tag, like "train", "val", or one of self._eval_tags.
gatr/experiments/nbody/experiment.py:25
Method
_make_data_loader
Creates a data loader. Parameters ---------- dataset : torch.nn.utils.data.Dataset Dataset. batch_size :
gatr/experiments/arteries/experiment.py:240
Method
_move_files
Move python files directly.
docker/ext_packages/install_upstream_python_packages.py:218
Method
build_attention_mask
Construct block-diagonal attention mask.
gatr/experiments/arteries/wrappers.py:66
Method
cast
Casts an object as SelfAttentionConfig.
gatr/layers/attention/config.py:100
Function
click_runner
Returns a CLI runner.
tests/conftest.py:8
Function
decorated_func
Decorated func.
gatr/utils/misc.py:231
Function
decorator
Decorator that casts input tensors to minimum precision.
gatr/utils/misc.py:206
Function
disable_opt_einsum
Disable usage of opt_einsum by torch during tests in this module.
tests/gatr/utils/test_einsum.py:23
Function
einsum_eq_fixture
Provides a non-trivial einsum equation.
tests/gatr/utils/test_einsum.py:31
Method
embed_into_ga
Embeds raw inputs into the geometric algebra (+ scalar) representation. To be implemented by subclasses. Parameters --------
tests_regression/regression_datasets/translation_norm.py:47
Method
embed_into_ga
Embeds raw inputs into the geometric algebra (+ scalar) representation. To be implemented by subclasses. Parameters --------
tests_regression/regression_datasets/translate_point.py:49
Method
embed_into_ga
Embeds raw inputs into the geometric algebra (+ scalar) representation. To be implemented by subclasses. Parameters --------
tests_regression/regression_datasets/points_distance.py:49
Method
embed_into_ga
Embeds raw artery data into the geometric algebra. We use the following parameterization of the artery mesh: - the node positions of
gatr/experiments/arteries/wrappers.py:76
Method
embed_into_ga
Embeds raw inputs into the geometric algebra (+ scalar) representation. Parameters ---------- inputs : torch.Tensor with shap
gatr/experiments/nbody/wrappers.py:75
Function
example_operands_fixture
Provides tensors for a non-trivial einsum equation.
tests/gatr/utils/test_einsum.py:38
← previous
next →
201–300 of 452, ranked by callers