normal-computing / thermoxLinks
Exact OU processes with JAX
☆48Updated 2 months ago
Alternatives and similar repositories for thermox
Users that are interested in thermox are comparing it to the libraries listed below
Sorting:
- Multiple dispatch over abstract array types in JAX.☆124Updated this week
- Riemannian Optimization Using JAX☆49Updated last year
- Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX☆41Updated last year
- Equivariant Steerable CNNs Library for Pytorch https://quva-lab.github.io/escnn/☆30Updated last year
- Code used by the "Clifford Group Equivariant Neural Networks" paper.☆83Updated last year
- Repository for <Lorentz-Equivariant Geometric Algebra Transformers for High-Energy Physics> (J. Spinner et al 2024)☆25Updated this week
- Matrix-free linear algebra in JAX.☆119Updated last week
- An implementation of ESM2 in Equinox+JAX☆26Updated 3 months ago
- Automatic Functional Differentiation in JAX☆72Updated 2 months ago
- A collection of graph neural networks implementations in JAX☆32Updated last year
- Stencil computations in JAX☆71Updated last year
- PySCF on IPU☆42Updated last year
- Implementation of Forward Laplacian algorithm in JAX☆76Updated last week
- diffusionjax is a simple and accessible diffusion models package in JAX☆45Updated 4 months ago
- ☆51Updated 2 years ago
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆34Updated 3 months ago
- Graph neural networks in JAX.☆67Updated 11 months ago
- The Energy Transformer block, in JAX☆56Updated last year
- Implementation of approximate free-energy minimization in PyTorch☆19Updated 3 years ago
- Running Jax in PyTorch Lightning☆102Updated 5 months ago
- ☆52Updated 2 years ago
- Flow-matching algorithms in JAX☆92Updated 9 months ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆45Updated 4 months ago
- Code for the book "The Elements of Differentiable Programming".☆86Updated 2 months ago
- A Wadler--Lindig pretty printer for Python☆39Updated 3 weeks ago
- Code for NeurIPS 2024 paper: "Noether's razor: Learning Conserved Quantities" by Tycho F. A. van der Ouderaa, Mark van der Wilk, Pim de H…☆10Updated 7 months ago
- ☆11Updated 5 months ago
- Turn jitted jax functions back into python source code☆22Updated 5 months ago
- JAX Arrays for human consumption☆91Updated last year
- Pytorch-like dataloaders for JAX.☆83Updated last week