Perceiver (transformer variant) implemented in JAX and Flax
☆13Mar 29, 2021Updated 5 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:
- 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 5 years ago
- This repository reproduces the results in the paper "How expressive are transformers in spectral domain for graphs?"(published in TMLR)☆10Jul 10, 2022Updated 4 years ago
- Jax implementation of the AdaHessian optimizer☆19Mar 11, 2021Updated 5 years ago
- JAX implementation of Graph Attention Networks☆13Jan 29, 2022Updated 4 years ago
- Supplementary Material to accompany the paper, DJ Warne, SA Sisson, C Drovandi (2019) Acceleration of expensive computations in Bayesian…☆13Oct 23, 2020Updated 6 years ago
- End-to-end encrypted cloud storage - Proton Drive • AdSpecial offer: 40% Off Yearly / 80% Off First Month. Protect your most important files, photos, and documents from prying eyes.
- Binding affinity prediction for drug discovery☆10Aug 28, 2025Updated last year
- Code to reproduce experiments in Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flows☆14May 23, 2024Updated 2 years ago
- ☆20Feb 24, 2023Updated 3 years ago
- Code for utilising VAE as means of doing exact MCMC inference in complex high-dimensional space☆14Jun 20, 2023Updated 3 years ago
- ☆12Oct 7, 2020Updated 6 years ago
- Nested lists published on GitHub.☆13Sep 7, 2022Updated 4 years ago
- [ICML 2023] Non-Uniform Neural Operator (NUNO)☆28Jul 3, 2023Updated 3 years ago
- Graph Convolutional Networks in JAX☆34Jan 3, 2021Updated 5 years ago
- Repository for the paper Tackling covariate shift with node-based Bayesian neural networks (ICML 2022 oral)☆11Jul 16, 2024Updated 2 years ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- generative language model training on top of the JAX and Huggingface 🤗☆10May 11, 2022Updated 4 years ago
- ☆16May 30, 2023Updated 3 years ago
- ☆21Mar 14, 2021Updated 5 years ago
- Gradient-informed particle MCMC methods☆12Jan 29, 2024Updated 2 years ago
- Gibbs sampling for expanded ensembles and replica exchange simulations☆11Jun 1, 2014Updated 12 years ago
- Python tools for Morse Smale Complex analysis and visualization☆14Nov 10, 2021Updated 4 years ago
- 🔩 It would be really good if we are to predict the chances of defect occurring in steel so that necessary precautions could be taken dur…☆12Jun 14, 2023Updated 3 years ago
- 트랜스포머 블록을 활용한 상품명 자연어처리 기반 카테고리 분류 모델☆10Dec 5, 2022Updated 3 years ago
- some code for linlog model simulation☆10Sep 16, 2021Updated 5 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Lightning-like training API for JAX with Flax☆45Dec 8, 2024Updated last year
- Python implementation of paper "AntisymmetricRNN: A Dynamical System View on Recurrent Neural Networks"☆15Aug 2, 2019Updated 7 years ago
- Python optimisation of atomistic ligand charges to maximize receptor binding affinity☆12Aug 3, 2020Updated 6 years ago
- Visualization of hidden layer activations of small multilayer perceptrons (MLPs)☆10Aug 26, 2024Updated 2 years ago
- Code for OGB competition☆12Feb 24, 2024Updated 2 years ago
- An implementation of squared neural families in PyTorch☆14Oct 22, 2024Updated 2 years ago
- A list of cool web radios☆32Sep 19, 2023Updated 3 years ago
- ☆16Apr 8, 2025Updated last year
- Python function that performs a balanced cut-tree of a SciPy linkage matrix☆13Jul 19, 2023Updated 3 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- FastPair: Data-structure for the dynamic closest-pair problem.☆13Oct 26, 2026Updated last week
- Re-scoring a set of docked ligands with off-the-shelf algorithms to assess utility in virtual screening☆11Oct 13, 2021Updated 5 years ago
- Auction Algorithm for Sparse Linear Assignment Problems☆13Mar 15, 2021Updated 5 years ago
- This repo will contain and gather libs and symbols working with LTSpiceIV. Directory structure will be tha same as lib dir in LTSpice to …☆12Apr 2, 2016Updated 10 years ago
- 👨💻 my solutions to interview questions☆10May 14, 2016Updated 10 years ago
- ☆16Feb 21, 2023Updated 3 years ago
- Compare molecular structures after energy minimization in various force fields.☆11Aug 22, 2020Updated 6 years ago