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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- This repository reproduces the results in the paper "How expressive are transformers in spectral domain for graphs?"(published in TMLR)☆11Jul 10, 2022Updated 3 years ago
- ☆37Jul 16, 2025Updated 8 months ago
- Jax, Flax, examples (ImageClassification, SemanticSegmentation, and more...)☆10May 10, 2025Updated 10 months ago
- Code for paper on automation of discovery and synthesis of targeted molecules: https://iopscience.iop.org/article/10.1088/2632-2153/abe80…☆19Sep 18, 2021Updated 4 years ago
- JAX implementation of Graph Attention Networks☆13Jan 29, 2022Updated 4 years ago
- DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Supplementary Material to accompany the paper, DJ Warne, SA Sisson, C Drovandi (2019) Acceleration of expensive computations in Bayesian…☆13Oct 23, 2020Updated 5 years ago
- Code for utilising VAE as means of doing exact MCMC inference in complex high-dimensional space☆14Jun 20, 2023Updated 2 years ago
- Client SDK for the Mobile Wallet Protocol☆22Jun 26, 2025Updated 9 months ago
- Code to reproduce experiments in Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flows☆13May 23, 2024Updated last year
- ☆12Oct 7, 2020Updated 5 years ago
- ☆20Feb 24, 2023Updated 3 years ago
- [ICML 2023] Non-Uniform Neural Operator (NUNO)☆26Jul 3, 2023Updated 2 years ago
- Probabilistic deep learning using JAX☆15Feb 8, 2025Updated last year
- Unified Free Energy Dynamics (UFED) simulations with OpenMM☆35Aug 20, 2025Updated 7 months ago
- DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- ☆16May 30, 2023Updated 2 years ago
- Graph Convolutional Networks in JAX☆33Jan 3, 2021Updated 5 years ago
- Nested lists published on GitHub.☆13Sep 7, 2022Updated 3 years ago
- Gradient-informed particle MCMC methods☆12Jan 29, 2024Updated 2 years ago
- ☆20Mar 14, 2021Updated 5 years ago
- ☆16Sep 8, 2025Updated 6 months ago
- Total Body and Brain 3D Segmentation in Fetal MRI scans☆25Jul 5, 2019Updated 6 years ago
- Code of PriorVAE: encoding spatial priors with variational autoencoders☆12Jul 13, 2023Updated 2 years ago
- 트랜스포머 블록을 활용한 상품명 자연어처리 기반 카테고리 분류 모델☆10Dec 5, 2022Updated 3 years ago
- DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Python implementation of paper "AntisymmetricRNN: A Dynamical System View on Recurrent Neural Networks"☆15Aug 2, 2019Updated 6 years ago
- ☆13Apr 7, 2024Updated last year
- Gibbs sampling for expanded ensembles and replica exchange simulations☆11Jun 1, 2014Updated 11 years ago
- Modular pipeline for design assistance. PYPi Package: https://pypi.org/project/vai-lab☆14Dec 22, 2023Updated 2 years ago
- Code for OGB competition☆12Feb 24, 2024Updated 2 years ago
- Python optimisation of atomistic ligand charges to maximize receptor binding affinity☆12Aug 3, 2020Updated 5 years ago
- An implementation of "metnet3" in Pytorch☆39Updated this week
- Software for dynamic spherical harmonic analysis☆10May 1, 2020Updated 5 years ago
- ☆14Dec 9, 2021Updated 4 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Examples on how to make use of DestinE Data Lake services☆14Updated this week
- A list of cool web radios☆32Sep 19, 2023Updated 2 years ago
- An implementation of squared neural families in PyTorch☆14Oct 22, 2024Updated last year
- Web Service wrapper for accessing the AmbiverseNLU KG stored in Neo4j☆12Nov 16, 2022Updated 3 years ago
- Minimal, fast + educational reimplementation of the TabICLv2 architecture☆59Feb 12, 2026Updated last month
- MoveFormer: a Transformer-based model for step-selection animal movement modelling☆12Aug 5, 2023Updated 2 years ago
- Minibatch loading for on-disk anndata files, co-developed by Lamin Labs and scverse☆27Updated this week