JAX + Attention Learn To Solve Routing Problems
Reinplementation of the paper Attention, Learn to Solve Routing Problems! using and Flax. Fully suitable for GPU and TPU execution.
Huge thanks to previous implementations in PyTorch and TensorFlow.
You might also like...
A High-precision Semantic Segmentation Method Combining Adversarial Learning and Attention Mechanism
A High-precision Semantic Segmentation Method Combining Adversarial Learning and Attention Mechanism
Implementation of Enformer, attention network for predicting gene expression, in Jax
Implementation of Enformer, attention network for predicting gene expression, in Jax
FedFormer: Contextual Federation with Attention in Reinforcement Learning (NeurIPS 2022)
FedFormer: Contextual Federation with Attention in Reinforcement Learning (NeurIPS 2022)
Object detection in remote sensing images using CNN and attention mechanism
Object detection in remote sensing images using CNN and attention mechanism
Fast Vision Transformers with HiLo Attention
Fast Vision Transformers with HiLo Attention
A simple cross attention that updates both the source and target in one step
A simple cross attention that updates both the source and target in one step
BoxeR: Box-Attention for 2D and 3D Transformers
BoxeR: Box-Attention for 2D and 3D Transformers
Implementation of Memorizing Transformers (ICLR 2022), attention net augmented with indexing and retrieval of memories using approximate nearest neigh
Implementation of Memorizing Transformers (ICLR 2022), attention net augmented with indexing and retrieval of memories using approximate nearest neigh