Graph Neural Networks with JAX/NNX
☆20Aug 12, 2026Updated 2 months ago
Alternatives and similar repositories for jraphx
Users that are interested in jraphx are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- OpenMM plugin for running JAX force fields through JaxForce☆16Aug 10, 2026Updated 2 months ago
- Euclid-equivariant operations for JAX☆27Oct 15, 2026Updated last week
- ☆12Apr 17, 2023Updated 3 years ago
- Python module for parsing, writing, aligning, and manipulating glycan structrures☆10Oct 7, 2026Updated 2 weeks ago
- This repo contains the code for the paper "Object-cropping for SSL".☆18Feb 14, 2023Updated 3 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Differentiable Electrochemistry Simulation Supporting Diffusion, Migration and Convection☆13Updated this week
- CSMC is a Python library for performing column subset selection in matrix completion tasks. It provides an implementation of the CSSMC me…☆12Jun 9, 2026Updated 4 months ago
- Benchmarking and validation of Machine Learning Interatomic Potentials (MLIP)☆19Sep 30, 2026Updated 3 weeks ago
- ☆13Oct 29, 2024Updated last year
- Improving Language Understanding from Screenshots. Paper: https://arxiv.org/abs/2402.14073☆32Jul 9, 2024Updated 2 years ago
- [TMLR 2025] Stability-Aware Training of Machine Learning Force Fields with Differentiable Boltzmann Estimators☆17Nov 20, 2025Updated 11 months ago
- ☆13Sep 29, 2023Updated 3 years ago
- Automatic Sparse Differentiation in JAX.☆59Updated this week
- R integration for edlib, a C/C++ library for pairwise sequence alignment using edit distance (Levenshtein distance).☆11Jul 20, 2025Updated last year
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Ready to run PyTorch implementation of Data2Vec 2.0: Highly efficient self-supervised representation learning for vision, speech and text…☆16Mar 29, 2023Updated 3 years ago
- org-mode templates for PhD hacking☆10Sep 4, 2025Updated last year
- Convert jax functions to sympy for symbolic analysis. This is useful for sparsity detection.☆22Jul 15, 2026Updated 3 months ago
- [ICML 2026] Direct 3D-Aware Object Insertion via Decomposed Visual Proxies☆44Jul 3, 2026Updated 3 months ago
- ☆22Sep 26, 2026Updated last month
- Fast and accurate tool for calculating Average Nucleotide Identity (ANI) among virus and bacteria genomes☆13Sep 12, 2026Updated last month
- volcanic source model and inversion framework in python☆23Apr 22, 2026Updated 6 months ago
- PEnG-motif is an open-source software package for searching statistically overrepresented motifs (position specific weight matrices, PWMs…☆10Nov 27, 2019Updated 6 years ago
- Scripts for finding restriction-modification systems in bacterial genomes.☆10Nov 9, 2023Updated 2 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Microbenchmarking hyperparameter tuning for JAX functions.☆24Oct 9, 2026Updated 2 weeks ago
- SParse lINear solvers in JAX☆59Aug 4, 2026Updated 2 months ago
- ☆11Feb 20, 2026Updated 8 months ago
- For bluntifying overlapped GFAs☆13Jul 26, 2024Updated 2 years ago
- Cython bindings and Python interface to Opal, a SIMD-accelerated database search aligner.☆11Mar 25, 2026Updated 7 months ago
- ☆16Aug 12, 2024Updated 2 years ago
- A Python Implementation of Predictive Coding (Whittington & Bogacz 2017)☆19Jul 28, 2020Updated 6 years ago
- Predictive coding networks for temporal prediction☆21Mar 20, 2024Updated 2 years ago
- Code for the preprint "Cache Me If You Can: How Many KVs Do You Need for Effective Long-Context LMs?"☆48Jul 29, 2025Updated last year
- 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.
- AI-native open-source finite-element platform connecting engineering, computation, data, and AI.☆30Updated this week
- NixOS port of basecamp/omarchy (Quattro) — vendored upstream, not rewritten☆35Updated this week
- Fast FracMinHash sketch generator☆11Jan 7, 2026Updated 9 months ago
- ☆20Jan 2, 2024Updated 2 years ago
- Data Tools for Interferometric Synthetic-Aperture Radar Satellite Data☆13Nov 7, 2018Updated 7 years ago
- [NeurIPS'25] RiverMamba: A State Space Model for Global River Discharge and Flood Forecasting☆21Sep 2, 2026Updated last month
- PolyFrame is a computational framework for 3D Graphic Statics and Reciporcal Polyhedral Frames☆12Jun 1, 2021Updated 5 years ago