Graph Learning with JAX
☆14Jul 11, 2022Updated 3 years ago
Alternatives and similar repositories for galax
Users that are interested in galax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆53Feb 27, 2023Updated 3 years ago
- Probabilistic structure discovery for rich relational systems☆14Jul 9, 2024Updated last year
- Implementation of Action Matching for the Schrödinger equation☆25Jun 18, 2023Updated 2 years ago
- Very deep VAEs in JAX/Flax☆47Jun 16, 2021Updated 4 years ago
- ☆32May 30, 2024Updated 2 years 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.
- ☆35Sep 12, 2022Updated 3 years ago
- Code repository for Particle Denoising Diffusion Sampler☆15Apr 2, 2024Updated 2 years ago
- Euclidean Wasserstein-2 optimal transportation☆48Aug 19, 2023Updated 2 years ago
- ☆33Jan 14, 2021Updated 5 years ago
- Learning the optimal transport map via input convex neural neworks☆44Jun 26, 2020Updated 5 years ago
- Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.☆331Mar 20, 2024Updated 2 years ago
- The uncompromising Jupyter notebook formatter.☆14Jan 26, 2024Updated 2 years ago
- LongSom tool for long-reads☆12May 20, 2025Updated last year
- Implementation of Action Matching☆52Jun 5, 2023Updated 2 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Library for Bayesian Neural Networks in PyTorch (first version as published in ProbProg2020)☆42Oct 5, 2021Updated 4 years ago
- 🧮 Extended Latent Dirichlet Allocation for Collaborative Filtering in Recommender Systems.☆42May 16, 2022Updated 4 years ago
- ☆12Apr 13, 2020Updated 6 years ago
- ☆85May 22, 2026Updated last week
- Model-based reinforcement learning in TensorFlow☆56Jul 27, 2021Updated 4 years ago
- Generative Modeling with Optimal Transport Maps - ICLR 2022☆61May 3, 2022Updated 4 years ago
- Inference in Bayesian Networks with R☆12Feb 6, 2023Updated 3 years ago
- Hückel model + JAX☆14Oct 13, 2022Updated 3 years ago
- Extended stochastic block models with application to criminal networks☆14Jan 16, 2022Updated 4 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- A set of Pyro models and functions to infer CNA from scRNA-seq data☆11Aug 14, 2023Updated 2 years ago
- Block coordinate descent for Sparse NMF☆14Oct 13, 2019Updated 6 years ago
- Normalizing Flows using JAX☆87Nov 30, 2023Updated 2 years ago
- Finding the causality in biological pathways☆14Jul 12, 2023Updated 2 years ago
- Composable kernels for scikit-learn implemented in JAX.☆47Oct 26, 2020Updated 5 years ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆105Oct 23, 2023Updated 2 years ago
- NotImplementedError: VJP of gammainc wrt argnum 0 not defined☆15Oct 15, 2020Updated 5 years ago
- A collection of graph neural networks implementations in JAX☆35Nov 28, 2023Updated 2 years ago
- ☆26Nov 25, 2024Updated 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.
- Map single-cell transcriptomes to copy number evolutionary trees.☆14Aug 18, 2024Updated last year
- Powerful and lightweight package to identify tissue compartments in spatial transcriptomics datasets.☆19Jun 9, 2025Updated 11 months ago
- ☆12Feb 13, 2022Updated 4 years ago
- ☆15Jun 22, 2022Updated 3 years ago
- Meta Optimal Transport☆108May 25, 2023Updated 3 years ago
- Movies Recommendation with Hierarchical Poisson Factorization in Edward☆18Jun 20, 2017Updated 8 years ago
- [ICLR 22] Value Gradient weighted Model-Based Reinforcement Learning.☆25Apr 15, 2023Updated 3 years ago