Graph Networks with Jax
☆11Mar 23, 2022Updated 4 years ago
Alternatives and similar repositories for grax
Users that are interested in grax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Steerable E(3) GNN in jax☆24Oct 1, 2023Updated 3 years ago
- Official repository of the paper "Pyramidal Reservoir Graph Neural Networks".☆10Mar 15, 2025Updated last year
- E(n) Equivariant GNN in jax☆14Aug 31, 2023Updated 3 years ago
- A Jax package to perform Neural Density Estimation from cosmological probes.☆13Jun 19, 2026Updated 4 months ago
- Exploring how ChatGPT can be used to accelerate research in cosmology.☆13Dec 12, 2022Updated 3 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- A collection of graph neural networks implementations in JAX☆36Nov 28, 2023Updated 2 years ago
- A Jax wrapper for cudaKDTree☆11Sep 26, 2025Updated last year
- Paper introducing jax-cosmo☆13Apr 27, 2023Updated 3 years ago
- Code for the paper "Towards Reliable Simulation-Based Inference with Balanced Neural Ratio Estimation".☆15Nov 14, 2022Updated 3 years ago
- PaiNN in jax☆11Jan 14, 2025Updated last year
- Stochastic trace estimation using JAX☆22Aug 27, 2026Updated 2 months ago
- Beginner-friendly serverless LLM deployment with Replicate & fly.io☆13Sep 3, 2023Updated 3 years ago
- 🤖 A streamlined, easy‑to‑install implementation of Fourier Neural Operators for time‑dependent and cloud‑point learning tasks.☆10May 28, 2023Updated 3 years ago
- JAX port of lenstronomy, for parallelized, GPU accelerated, and differentiable gravitational lensing and image simulations.☆21Updated this week
- 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.
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆21Mar 10, 2024Updated 2 years ago
- Deep Neural Network Optimization Platform with Gradient-based, Gradient-Free Algorithms☆11Jan 13, 2020Updated 6 years ago
- Vector Quantile Regression☆20Apr 3, 2025Updated last year
- A simple implementation of Hamiltonian Monte Carlo in JAX.☆20Feb 8, 2024Updated 2 years ago
- An experiment: emcee implemented in JAX☆28Jul 1, 2022Updated 4 years ago
- ☆11Nov 11, 2024Updated last year
- ACM_exercise code for OVaOJ connecting to 算法竞赛入门训练指南☆11Aug 18, 2014Updated 12 years ago
- ☆13Mar 1, 2026Updated 7 months ago
- A semi-analytical satellite galaxy and dark matter halo generator☆31Feb 26, 2024Updated 2 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- SymPy based framework for optimized code generation for BSSN formulation of Einstein equation for heterogeneous platforms.☆11Updated this week
- [ICML 2026] The official code repo for our paper "Neural QAOA²: Differentiable Joint Graph Partitioning and Parameter Initialization for …☆19May 18, 2026Updated 5 months ago
- Scripts and data deposition for de novo heterodimers manuscript☆11Dec 15, 2018Updated 7 years ago
- xc deep neural network (XCNN) with a differentiable DFT☆11Jan 19, 2022Updated 4 years ago
- ☆12May 27, 2021Updated 5 years ago
- Snippets for competitive programming in Haskell☆21Sep 3, 2017Updated 9 years ago
- Converts a FigJam flowchart to json☆12Aug 4, 2024Updated 2 years ago
- Learning astrophysics with top-k sparse feature networks (SFNets)☆11Feb 21, 2026Updated 8 months ago
- ☆15Jul 12, 2026Updated 3 months ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- Neural Simulation-based Inference with GNN for Jeans Modeling☆10Feb 23, 2024Updated 2 years ago
- Tools for training and evaluating CIDER functionals for use in Density Functional Theory calculations☆10Mar 5, 2024Updated 2 years ago
- Stainless neural networks in JAX☆37Jun 13, 2026Updated 4 months ago
- Differentiable strong lens modelling on CPUs and GPUs☆32Jul 8, 2026Updated 3 months ago
- Dynamic Flux Balance analysis in Python☆12Jan 22, 2016Updated 10 years ago
- 🚀 A powerful library for efficient training of Neural Fields at scale.☆31Jan 22, 2024Updated 2 years ago
- Software tool for metabolic modelling in Systems Biology☆10Jul 24, 2026Updated 3 months ago