Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Extensions to relative positional encoding #203

Merged
merged 22 commits into from
Mar 18, 2022

Conversation

cswinter
Copy link
Collaborator

@cswinter cswinter commented Mar 17, 2022

Adds several new features to relative positional encoding that allow it to work better in continuous spaces:

  • adding a linear projection of the relative position/distance to the relative keys/queries
  • partition functions for relative positional encoding based on the relative angle, distance, or polar coordinates
  • interpolating between partitions
  • (partially implemented) rotating relative positions by the orientation of each entity

Other changes not directly related to relpos encoding:

  • improved xprun builds
  • new "webpdb" feature to catch crashes and allow for debugging with webpdb
  • misc bug fixes

@cswinter cswinter enabled auto-merge (squash) March 18, 2022 00:05
@cswinter cswinter merged commit 35c920b into main Mar 18, 2022
@cswinter cswinter deleted the clemens/better-relpos-encoding branch March 18, 2022 00:07
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

1 participant