Alternatives and similar repositories for VN-transformer:

Users that are interested in VN-transformer are comparing it to the libraries listed below

• lucidrains / gradnorm-pytorch
  A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch
  ☆85Updated last year
• autonomousvision / gta
  [ICLR'24] GTA: A Geometry-Aware Attention Mechanism for Multi-view Transformers
  ☆121Updated 8 months ago
• lucidrains / coordinate-descent-attention
  Implementation of an Attention layer where each head can attend to more than just one token, using coordinate descent to pick topk
  ☆46Updated last year
• lucidrains / spline-based-transformer
  Implementation of the proposed Spline-Based Transformer from Disney Research
  ☆86Updated 3 months ago
• lucidrains / kalman-filtering-attention
  Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"
  ☆57Updated last year
• yklcs / jaxsplat
  3D Gaussian Splatting in JAX
  ☆55Updated 8 months ago
• facebookresearch / riemannian-fm
  code for "Riemannian Flow Matching on General Geometries".
  ☆200Updated 11 months ago
• facebookresearch / replay_dataset
  Download scripts and tools for Replay dataset.
  ☆31Updated last year
• lebellig / discrete-fm
  Educational implementation of the Discrete Flow Matching paper
  ☆76Updated 5 months ago
• FlyingGiraffe / vnn-pc
  Vector Neuron pointcloud networks for classification and segmentation. Separate training setups for VN-DGCNN and VN-PointNet
  ☆31Updated 2 years ago
• lebellig / flow-matching
  Annotated Flow Matching paper
  ☆172Updated 5 months ago
• jw9730 / setvae
  [CVPR'21] SetVAE: Learning Hierarchical Composition for Generative Modeling of Set-Structured Data, in PyTorch
  ☆74Updated 2 years ago
• lucidrains / self-reasoning-tokens-pytorch
  Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto
  ☆55Updated 9 months ago
• vsitzmann / neural-isometries
  Official JAX implementation of neural isometries - taming transformations for equivariant ML
  ☆34Updated 6 months ago
• bm371613 / conditional-flow-matching
  ☆14Updated 9 months ago
• mihirp1998 / Slot-TTA
  Slot-TTA shows that test-time adaptation using slot-centric models can improve image segmentation on out-of-distribution examples.
  ☆26Updated last year
• yilundu / gem
  [NeurIPS 2021] Code for Learning Signal-Agnostic Manifolds of Neural Fields
  ☆67Updated last year
• google-deepmind / functa
  ☆154Updated 6 months ago
• lucidrains / gotennet-pytorch
  Implementation of GotenNet, new SOTA 3d equivariant transformer, in Pytorch
  ☆54Updated 2 weeks ago
• yilundu / ired_code_release
  ☆51Updated 8 months ago
• lucidrains / equiformer-pytorch
  Implementation of the Equiformer, SE3/E3 equivariant attention network that reaches new SOTA, and adopted for use by EquiFold for protein…
  ☆256Updated 2 months ago
• lucidrains / einops-exts
  Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️
  ☆53Updated 2 years ago
• apple / ml-agm
  ☆44Updated 9 months ago
• louaaron / Reflected-Diffusion
  [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)
  ☆157Updated last year
• YangLing0818 / consistency_flow_matching
  Official Implementation for "Consistency Flow Matching: Defining Straight Flows with Velocity Consistency"
  ☆192Updated last month
• Jhomanik / Optimal-Flow-Matching
  The official repository for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step" (NeurIPS 2024)
  ☆63Updated 2 months ago
• locuslab / ImpSq
  Implicit^2: Implicit model for implicit neural representations
  ☆28Updated 3 years ago
• mbreuss / consistency_models_toy_task
  Unofficial minimal implementation of consistency models (CM) proposed by Song et al. 2023 on a 1D toy task in pytorch
  ☆19Updated last year
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆49Updated 6 months ago