sooheon / perceiver-jaxView external linksLinks
Perceiver (transformer variant) implemented in JAX and Flax
☆13Mar 29, 2021Updated 4 years ago
Alternatives and similar repositories for perceiver-jax
Users that are interested in perceiver-jax are comparing it to the libraries listed below
Sorting:
- JAX implementation of Graph Attention Networks☆13Jan 29, 2022Updated 4 years ago
- ☆29Jul 16, 2025Updated 6 months ago
- Python implementation of paper "AntisymmetricRNN: A Dynamical System View on Recurrent Neural Networks"☆15Aug 2, 2019Updated 6 years ago
- ☆21Mar 14, 2021Updated 4 years ago
- A Jax/Stax implementation of the general meta learning paper: Oh, J., Hessel, M., Czarnecki, W.M., Xu, Z., van Hasselt, H.P., Singh, S. a…☆23Dec 22, 2020Updated 5 years ago
- [ICML 2023] Non-Uniform Neural Operator (NUNO)☆26Jul 3, 2023Updated 2 years ago
- ☆12Oct 7, 2020Updated 5 years ago
- 트랜스포머 블록을 활용한 상품명 자연어처리 기반 카테고리 분류 모델☆10Dec 5, 2022Updated 3 years ago
- Graph Convolutional Networks in JAX☆33Jan 3, 2021Updated 5 years ago
- Unified Free Energy Dynamics (UFED) simulations with OpenMM☆35Aug 20, 2025Updated 5 months ago
- Examples on how to make use of DestinE Data Lake services☆14Updated this week
- A tutorial on JAX (https://github.com/google/jax/)☆47Jan 16, 2019Updated 7 years ago
- ☆14Mar 3, 2022Updated 3 years ago
- This repository reproduces the results in the paper "How expressive are transformers in spectral domain for graphs?"(published in TMLR)☆12Jul 10, 2022Updated 3 years ago
- Water wave models in one dimension☆10Jan 4, 2026Updated last month
- Learning Generalized Physical Representation from a Few Examples☆16Updated this week
- Auction Algorithm for Sparse Linear Assignment Problems☆13Mar 15, 2021Updated 4 years ago
- The Maveric project is the source of the RIC Algorithm Development Platform (RADP). It enables the development and evaluation of dynamic …☆16Nov 6, 2025Updated 3 months ago
- Selective Copying Task with Mamba Model. This repository contains a simple implementation for reproducing the selective copying task with…☆12Jun 3, 2024Updated last year
- Source code and datasets for Circuit Design Completion using GNNs paper☆10Jan 26, 2023Updated 3 years ago
- This project focuses on using deep learning to replace text in images while retaining the same font and style.☆10Dec 9, 2019Updated 6 years ago
- A Cython library to solve the Bittensor registration POW on CUDA☆15Aug 15, 2025Updated 5 months ago
- Code for utilising VAE as means of doing exact MCMC inference in complex high-dimensional space☆14Jun 20, 2023Updated 2 years ago
- An implementation of "metnet3" in Pytorch☆37Jan 30, 2026Updated 2 weeks ago
- Probabilistic deep learning using JAX☆15Feb 8, 2025Updated last year
- Code for Max-Margin Deep Generative Models☆12Jan 1, 2015Updated 11 years ago
- Electron-based Virtual MIDI Controller that like Novation Launch Control☆10Oct 28, 2016Updated 9 years ago
- Reproduce analyses and baselines from the State paper☆13Oct 24, 2025Updated 3 months ago
- Layered distributions using FLAX/JAX☆10Dec 13, 2020Updated 5 years ago
- ☆12Aug 24, 2020Updated 5 years ago
- Coq集合模型论☆11Aug 18, 2022Updated 3 years ago
- ☆13Dec 3, 2022Updated 3 years ago
- The Click example that I use in my tutorial articles.☆13May 23, 2018Updated 7 years ago
- A new language for optimization☆13May 17, 2021Updated 4 years ago
- Modular pipeline for design assistance. PYPi Package: https://pypi.org/project/vai-lab☆14Dec 22, 2023Updated 2 years ago
- Minibatch loading for on-disk anndata files, co-developed by Lamin Labs and scverse☆24Updated this week
- This network estimation procedure combines l1-regularized logistic regression with model selection based on the Extended Bayesian Informa…☆10Oct 12, 2023Updated 2 years ago
- Vue.js wrapper over canvas manipulation library fabric.js — **Note: not quite ready for prime time**☆11Nov 27, 2016Updated 9 years ago
- [CogSci'21] Study of human inductive biases in CNNs and Transformers.☆43May 18, 2021Updated 4 years ago