Implementation of a Set Transformer in JAX from the paper 'Set Transformer: A Framework for Attention-based Permutation-Invariant Neural Networks' by Lee et al. (2019)
-
Updated
Jul 24, 2024 - Jupyter Notebook
#
set-transformer
Here are 2 public repositories matching this topic...
Fast Amortized Neural Symbolic Regression with Transformers and SimpliPy
python machine-learning natural-language-processing transformer symbolic-regression interpretable-ml explainable-ai equation-discovery set-transformer amortized-symbolic-regression
-
Updated
Dec 12, 2025 - Python
Improve this page
Add a description, image, and links to the set-transformer topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the set-transformer topic, visit your repo's landing page and select "manage topics."