Xtructure is datastructure for using in JAX
☆21Mar 3, 2026Updated 2 weeks ago
Alternatives and similar repositories for Xtructure
Users that are interested in Xtructure are comparing it to the libraries listed below
Sorting:
- Jaxpr Visualisation Tool☆36Dec 22, 2024Updated last year
- JAX Multi-Agent RL, Neuro-Evolution, and A-Life Library☆13Oct 12, 2025Updated 5 months ago
- SParse lINear solvers in JAX☆37Feb 22, 2026Updated last month
- A Wadler–Lindig pretty printer for Python☆46Jan 16, 2026Updated 2 months ago
- Awesome Equinox - A curated list of resources of https://github.com/patrick-kidger/equinox☆16Jun 24, 2024Updated last year
- Add a tqdm progress bar to your JAX scans and loops.☆125May 9, 2025Updated 10 months ago
- The official implementations of Intention-conditioned Flow Occupancy Models (InFOM)☆31Feb 5, 2026Updated last month
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆40Jan 16, 2026Updated 2 months ago
- Unitful Quantities in JAX☆57Mar 13, 2026Updated last week
- Code for the paper "Stable Gradients for Stable Learning at Scale in Deep Reinforcement Learning". Great performance in many environments…☆38Oct 24, 2025Updated 4 months ago
- ☆12Jun 17, 2025Updated 9 months ago
- PyTinyRenderer is a simple C++/CPU renderer, this fork adds Python bindings and triangle clipping and OpenGL compatible view/projection m…☆25Oct 1, 2024Updated last year
- An innovative Verilog-A compiler - reloaded☆41Feb 26, 2026Updated 3 weeks ago
- Efficient Differentiable n-d PDE Solvers in JAX.☆165Mar 2, 2026Updated 2 weeks ago
- Differentiable collision detection for capsules in JAX.☆52Dec 2, 2023Updated 2 years ago
- Tidy autoregressive inference in JAX☆15Sep 1, 2025Updated 6 months ago
- BONNI: Bayesian Optimization via Neural Network surrogates and Interior Point Optimization☆28Feb 20, 2026Updated last month
- e-Stellarator Simulation and Optimization Suite☆14Mar 13, 2026Updated last week
- A MkDocs plugin for embedding interactive jupyter notebooks in your docs via jupyterlite.☆25Dec 18, 2025Updated 3 months ago
- Simple JAX Graphics Library.☆37Nov 3, 2024Updated last year
- JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library☆124Updated this week
- Efficient SDE samplers including Gaussian-based probabilistic solvers. Written in JAX.☆10Feb 8, 2025Updated last year
- LiteX based White Rabbit PCIe NIC developped for Warsaw University of Technology.☆21Mar 3, 2026Updated 2 weeks ago
- Kinematic and dynamic models of continuum and articulated soft robots.☆16Nov 22, 2025Updated 4 months ago
- Prototype for sparse automatic differentiation in JAX + Julia☆21Jun 17, 2025Updated 9 months ago
- Differentiable Rasteriser implemented in JAX. Reference: https://github.com/erwincoumans/tinyrenderer, https://github.com/ssloy/tinyrende…☆79Nov 14, 2023Updated 2 years ago
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆274Sep 22, 2025Updated 6 months ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆264Oct 31, 2025Updated 4 months ago
- Minimal, lightweight JAX implementations of popular models.☆214Mar 4, 2026Updated 2 weeks ago
- Bounding Interval Hierarchy and Bounding Volume Hierarchy for Node.js☆23Apr 15, 2018Updated 7 years ago
- JAX-accelerated nuclear equation of state inference and TOV solvers☆10Mar 13, 2026Updated last week
- Reconstructing tomography data. Faster!☆15Dec 17, 2025Updated 3 months ago
- What's the simplest Turing Machine with unknown behavior?☆13Jun 18, 2016Updated 9 years ago
- Schedule free optimiser implemented in JAX using Optimistix☆15May 29, 2024Updated last year
- Next generation MCMC samplers with automatic differentiaion and adaptive Poisson thinning☆13Mar 9, 2026Updated last week
- Tutorial to learn basic features of atomate2☆15Sep 17, 2024Updated last year
- ☆19Jan 25, 2026Updated last month
- Functional algorithms - definitions and implementations☆13Oct 17, 2025Updated 5 months ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆75Feb 12, 2026Updated last month