☆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:
- Unofficial JAX implementation of the SOAP optimizer (https://arxiv.org/abs/2409.11321)☆29Jul 21, 2026Updated 2 months ago
- 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
- Matrix-free linear algebra in JAX.☆180Sep 2, 2026Updated 3 weeks ago
- Efficient SDE samplers including Gaussian-based probabilistic solvers. Written in JAX.☆10Feb 8, 2025Updated last year
- Official Pytorch implementation for "Tree-Based Diffusion Schrödinger Bridge with Applications to Wasserstein Barycenters" (NeurIPS 2023)☆10Mar 5, 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.
- Convolutions and more as einsum for PyTorch☆18Jun 6, 2024Updated 2 years ago
- This repo contains papers, books, tutorials and resources on Riemannian optimization.☆66Sep 11, 2026Updated last week
- Automatically exported from code.google.com/p/another-chebpy☆13Apr 28, 2015Updated 11 years ago
- Julia-themed Quarto.☆11Jul 10, 2024Updated 2 years ago
- This repository consists of two-step training DeepONet code.☆13May 22, 2024Updated 2 years ago
- A demo code for implementation of differentiable thermodynamic modeling in JAX.☆11Sep 18, 2021Updated 5 years ago
- A package for computing matrix exponentials and finite horizon Gramians☆11Jan 21, 2026Updated 8 months ago
- 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
- DM i AI 2022☆10Oct 4, 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.
- Simulation of Systems of interacting mean-field Particles with High Efficiency☆16Nov 27, 2023Updated 2 years ago
- Functional differential geometry in Julia☆12Oct 20, 2022Updated 3 years ago
- Fast & stable PRNG☆11Dec 17, 2021Updated 4 years ago
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Jul 19, 2023Updated 3 years ago
- Superpixels through Iterative CLEarcutting (SICLE) framework☆11Aug 25, 2023Updated 3 years ago
- Discretization of the Lippmann--Schwinger equation with periodic boundary conditions☆13Updated this week
- ☆15Nov 18, 2024Updated last year
- Example code to speed up GP inference with gradients for high-dimensional inputs.☆14Jul 12, 2021Updated 5 years ago
- 🎓 Ph.D. thesis: Towards realistic numerical simulations of Majorana devices☆16Aug 13, 2020Updated 6 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.
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆15May 8, 2023Updated 3 years ago
- Examples using petsc4py, cython and petsc to solve Finite Volume Problems.☆14Sep 10, 2017Updated 9 years ago
- Memory and disk memoizer with compression support☆32May 19, 2026Updated 4 months ago
- Spectral Graph Attention Network with Fast Eigen-approximation☆11Dec 24, 2021Updated 4 years ago
- a curvature model for the generation of tubes☆19Aug 17, 2022Updated 4 years ago
- Latent Mamba Operator for Partial Differential Equations☆17Jun 3, 2025Updated last year
- This is the code of our work Are More Layers Beneficial to Graph Transformers? published on ICLR 2023.☆22May 27, 2023Updated 3 years ago
- Accompanying code for "An Elastic Basis for Spectral Shape Correspondence"☆12Aug 2, 2023Updated 3 years ago
- Stencil computations in JAX☆73Nov 15, 2025Updated 10 months 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.
- GPU-accelerated differentiable sparse linear solver library for JAX☆52Aug 30, 2026Updated 3 weeks ago
- Meta-Learning an In-Context Transformer Model of Human Higher Visual Cortex (NeurIPS 2025)☆18Aug 6, 2026Updated last month
- ☆20Aug 27, 2024Updated 2 years ago
- A tool for laying out figures with geometric constraints☆20Dec 5, 2025Updated 9 months ago
- Source code for my PhD thesis: Backpropagation Beyond the Gradient☆21Feb 25, 2023Updated 3 years ago
- ☆20Dec 31, 2023Updated 2 years ago
- ☆18Oct 16, 2022Updated 3 years ago