SYMBOL INDEX (25 symbols across 1 files)

FILE: screenai/main.py
  function exists (line 20) | def exists(val):
  function default (line 24) | def default(val, d):
  function pair (line 28) | def pair(val):
  function divisible_by (line 32) | def divisible_by(numer, denom):
  function dynamic_patching (line 36) | def dynamic_patching(x, patch_size, image_size):
  function pad_dim_to (line 58) | def pad_dim_to(t, length, dim=0):
  function all_gather_variable_batch (line 64) | def all_gather_variable_batch(t):
  class AllGather (line 102) | class AllGather(Function):
    method forward (line 104) | def forward(ctx, x):
    method backward (line 111) | def backward(ctx, grads):
  class EmbedToLatents (line 127) | class EmbedToLatents(nn.Module):
    method __init__ (line 128) | def __init__(self, dim, dim_latents):
    method forward (line 132) | def forward(self, x):
  class CrossAttention (line 141) | class CrossAttention(nn.Module):
    method __init__ (line 155) | def __init__(
    method forward (line 197) | def forward(self, x, context):
  class MultiModalEncoder (line 249) | class MultiModalEncoder(nn.Module):
    method __init__ (line 270) | def __init__(
    method forward (line 297) | def forward(self, x: Tensor) -> Tensor:
  class MultiModalDecoder (line 316) | class MultiModalDecoder(nn.Module):
    method __init__ (line 337) | def __init__(
    method forward (line 368) | def forward(self, x: Tensor) -> Tensor:
  class ScreenAI (line 376) | class ScreenAI(nn.Module):
    method __init__ (line 412) | def __init__(
    method forward (line 487) | def forward(self, text: Tensor, img: Tensor) -> Tensor: