☆20Sep 5, 2024Updated 2 years ago
Alternatives and similar repositories for jaxgeometry
Users that are interested in jaxgeometry are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Tree traversals using JAX☆11Jun 2, 2026Updated 5 months ago
- ☆14Feb 23, 2025Updated last year
- This repository is the official implementation of "GraphMoRE: Mitigating Topological Heterogeneity via Mixture of Riemannian Experts" acc…☆12Sep 21, 2025Updated last year
- Unofficial JAX implementation of the SOAP optimizer (https://arxiv.org/abs/2409.11321)☆29Jul 21, 2026Updated 3 months ago
- [ NeurIPS 2023 ] Official Codebase for "Aligning Synthetic Medical Images with Clinical Knowledge using Human Feedback"☆20Oct 19, 2023Updated 3 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- A unified interface to compute hydrodynamic displacements.☆14Aug 18, 2026Updated 2 months ago
- Visualization of hidden layer activations of small multilayer perceptrons (MLPs)☆10Aug 26, 2024Updated 2 years ago
- Matrix-free linear algebra in JAX.☆183Oct 22, 2026Updated last week
- Time2Vec neural network components. From paper: "Time2Vec: Learning a Vector Representation of Time" - https://arxiv.org/pdf/1907.05321.p…☆10Nov 25, 2024Updated last year
- ☆13Jun 10, 2025Updated last year
- LaTeX style file for Transactions on Machine Learning Research☆15Jun 30, 2023Updated 3 years ago
- Official Pytorch implementation for "Tree-Based Diffusion Schrödinger Bridge with Applications to Wasserstein Barycenters" (NeurIPS 2023)☆10Mar 5, 2024Updated 2 years ago
- An inverse design framework combining SDF representations with a classifier-free guided diffusion model for one-shot synthesis of functio…☆11Jul 29, 2026Updated 3 months ago
- ☆13Mar 21, 2024Updated 2 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.
- Repository for the paper "Bayesian Quadrature on Riemannian Data Manifolds"☆10Jun 18, 2024Updated 2 years ago
- A package for computing matrix exponentials and finite horizon Gramians☆11Jan 21, 2026Updated 9 months ago
- Parser for bibliographic formats, including BibTeX, in pure Julia☆14Updated this week
- Code accompanying the NeurIPS 2021 Paper: A Probabilistic State Space Model for Joint Inference from Differential Equations and Data (Sch…☆13Nov 7, 2022Updated 3 years ago
- A demo code for implementation of differentiable thermodynamic modeling in JAX.☆11Sep 18, 2021Updated 5 years ago
- Functional differential geometry in Julia☆12Oct 20, 2022Updated 4 years ago
- Convolutions and more as einsum for PyTorch☆18Jun 6, 2024Updated 2 years ago
- Fast & stable PRNG☆11Dec 17, 2021Updated 4 years ago
- A small, composable interface for iterative algorithms in Julia☆13Oct 15, 2026Updated 2 weeks ago
- 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.
- Binaries for mathematicians☆10Mar 19, 2025Updated last year
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆15May 8, 2023Updated 3 years ago
- This repository contains the code of the paper "Riemannian dimensionality reduction for Symmetric Positive Definite Matrices".☆10Feb 6, 2025Updated last year
- A list of links to useful code snippets for manim☆14Jul 27, 2024Updated 2 years ago
- Superpixels through Iterative CLEarcutting (SICLE) framework☆11Oct 7, 2026Updated 3 weeks ago
- Automatic Sparse Differentiation in JAX.☆67Oct 21, 2026Updated last week
- Discretization of the Lippmann--Schwinger equation with periodic boundary conditions☆13Sep 26, 2026Updated last month
- Manifolds for the natural parameters of members of the exponential family of distributions☆11Sep 29, 2026Updated last month
- Differentiation on manifolds☆13Jul 20, 2026Updated 3 months ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Meta-Learning an In-Context Transformer Model of Human Higher Visual Cortex (NeurIPS 2025)☆18Aug 6, 2026Updated 2 months ago
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆18Jul 19, 2023Updated 3 years ago
- This repository consists of two-step training DeepONet code.☆13May 22, 2024Updated 2 years ago
- Examples using petsc4py, cython and petsc to solve Finite Volume Problems.☆14Sep 10, 2017Updated 9 years ago
- Latent Mamba Operator for Partial Differential Equations☆17Jun 3, 2025Updated last year
- Accompanying code for "An Elastic Basis for Spectral Shape Correspondence"☆12Aug 2, 2023Updated 3 years ago
- ☆10Jul 11, 2014Updated 12 years ago