A framework for GPU-friendly implementations of tree algorithms -- such as kNN and FoF -- in jax (with a CUDA backend)
☆49Jun 30, 2026Updated 2 weeks ago
Alternatives and similar repositories for jztree
Users that are interested in jztree are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- DISCO-DJ: DIfferentiable Simulations for COsmology - Done with Jax☆27May 16, 2026Updated 2 months ago
- e-Stellarator Simulation and Optimization Suite☆17Updated this week
- A demo code for implementation of differentiable thermodynamic modeling in JAX.☆11Sep 18, 2021Updated 4 years ago
- Jaxpr Visualisation Tool☆37Dec 22, 2024Updated last year
- Toolbox for single-cell data analysis on sample level☆25Jul 14, 2026Updated last week
- 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.
- ☆17May 21, 2026Updated 2 months ago
- Pure JAX implementation of the Non-Uniform Fast Fourier Transform (NUFFT)☆29Jul 2, 2026Updated 2 weeks ago
- Single-cell differential expression☆19Apr 23, 2026Updated 2 months ago
- Multi-scale initial conditions for cosmological simulations, v2☆22Aug 25, 2025Updated 10 months ago
- Unitful Quantities in JAX☆65Updated this week
- Fast array utilities☆15Updated this week
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆82Feb 12, 2026Updated 5 months ago
- Illico is a python library performing fast and lightweight wilcoxon rank-sum tests for single-cell RNASeq.☆20Jul 7, 2026Updated last week
- An automated enhanced sampling generation of training sets for chemically reactive machine learning interatomic potentials☆23Jun 6, 2026Updated last month
- 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.
- Vote on whether you think predicted crystal structures could be synthesised☆18Jul 29, 2024Updated last year
- Implementation of multiplex Leiden for analysis of spatial omics data☆22Jul 7, 2026Updated last week
- Differentiable Virtual Telescope to produce Mock Galaxy Images☆21Jul 13, 2026Updated last week
- HuBMAP Data Portal front end☆13Updated this week
- physics-informed neural network based method for solving single-cell population dynamics☆17Jun 23, 2026Updated 3 weeks ago
- ☆25Updated this week
- A fast, multi-target voronoi tessellator using the SYCL framework☆22Jul 15, 2025Updated last year
- The Stepped-Pressure Equilibrium Code, an advanced MRxMHD equilibrium solver.☆33Jun 23, 2026Updated 3 weeks ago
- A python package to process Direct Numerical Simulations☆22Jun 19, 2026Updated last month
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- An AnnData-based framework for integrated proteomics analysis☆28Jul 8, 2026Updated last week
- JAX implementation and extension of the healpy library☆15Jul 7, 2026Updated last week
- A simple JAX-powered simulation library for numerical experiments of fuzzy dark matter, stars, gas + more!☆49Jul 6, 2026Updated 2 weeks ago
- R API and htmlwidget for Vitessce☆45Aug 13, 2025Updated 11 months ago
- A Python assistant for performing and analyzing molecular dynamics simulations of soft matter systems☆22Aug 20, 2025Updated 11 months ago
- DISCnetMachineLearningCourse☆10Jul 2, 2025Updated last year
- A MkDocs plugin for embedding interactive jupyter notebooks in your docs via jupyterlite.☆25Jul 3, 2026Updated 2 weeks ago
- encore: Efficient isotropic 2-, 3- and 4-point correlation functions in C++ and CUDA☆21Sep 4, 2025Updated 10 months ago
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆60May 19, 2026Updated 2 months 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.
- High-Performance Sparse Matrix Library☆19Nov 23, 2025Updated 7 months ago
- MESHOID: MESHless Operations such as Integrals and Derivatives☆32Apr 13, 2026Updated 3 months ago
- [ICML 2025 GenBio Workshop] Official Implementation for "Electrostatics from Laplacian Eigenbasis for Neural Network Interatomic Potentia…☆18Jun 12, 2025Updated last year
- cinema toolkit for large data analysis and visualization☆14Sep 14, 2022Updated 3 years ago
- The AREPO visualization toolkit, a ray-tracing Voronoi tessellation volume renderer.☆59Aug 4, 2021Updated 4 years ago
- ☆19Jul 13, 2026Updated last week
- A python script to solve the Cahn-Hilliard equation using an implicit pseudospectral method☆55Jul 17, 2024Updated 2 years ago