Execute + differentiate Tesseracts as part of JAX programs, with full support for function transformations like JIT, grad, and more. âš¡
☆28Feb 24, 2026Updated last week
Alternatives and similar repositories for tesseract-jax
Users that are interested in tesseract-jax are comparing it to the libraries listed below
Sorting:
- Universal, autodiff-native software components for Simulation Intelligence. 📦☆95Feb 27, 2026Updated last week
- Jatmos is a scalable simulator for atmospheric dynamics, written in JAX.☆35Apr 21, 2025Updated 10 months ago
- Astrostatistics and Machine Learning class for the MSc degree in Astrophysics at the University of Milan-Bicocca (Italy)☆15Aug 30, 2025Updated 6 months ago
- KENN - A Transformer Encoder for Gravitational Waves☆11Sep 29, 2025Updated 5 months ago
- Source code repository for the AISTAT 2023 paper Transport Reversible Jump Proposals.☆10Mar 3, 2023Updated 3 years ago
- Superphot+, upgraded version of Superphot that uses nested sampling☆12Dec 31, 2025Updated 2 months ago
- Framework for computing kilonova light curves☆12Sep 2, 2025Updated 6 months ago
- Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.☆115Updated this week
- ☆10Oct 23, 2024Updated last year
- Code to reproduce experiments in Markovian Flow Matching: Accelerating MCMC with Continuous Normalizing Flows☆13May 23, 2024Updated last year
- Numerical relativity surrogate waveform in Jax☆19Aug 14, 2025Updated 6 months ago
- A vim plugin that lets you search with multiple tags.☆11Oct 1, 2020Updated 5 years ago
- JAX code for modeling nearly-Keplerian orbits☆17Jan 8, 2026Updated last month
- Codes to reproduce results in the paper☆10Apr 9, 2021Updated 4 years ago
- Octax: Accelerated CHIP-8 Arcade Environments for JAX☆38Feb 18, 2026Updated 2 weeks ago
- ☆10Jan 18, 2022Updated 4 years ago
- Extended material of the summer course "Introduction to High-Performance Computing in Python for Scientists!". Goethe Research Academy fo…☆11Jul 25, 2022Updated 3 years ago
- ☆11Jul 25, 2021Updated 4 years ago
- Rescure useful code in TensorNetwork by google for further personal usage.☆13Dec 6, 2024Updated last year
- ☆15Dec 11, 2025Updated 2 months ago
- PyBlastAfterglow: fast modeling of GRB and Kilonova afterglows☆12Jan 16, 2025Updated last year
- ☆10Nov 14, 2024Updated last year
- A fast, differentiable, and extensible public BBN code☆15Feb 13, 2026Updated 3 weeks ago
- Computational solid mechanics made easy with Jax☆50Oct 19, 2025Updated 4 months ago
- A lightweight RL environment for query optimization.☆16Sep 13, 2024Updated last year
- Extensible library for finite element programming☆14Mar 7, 2021Updated 4 years ago
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆13Jun 12, 2024Updated last year
- Corner plots -- now with ads!☆23Mar 31, 2025Updated 11 months ago
- Stochastic Gradient MCMC for Jax☆18May 19, 2025Updated 9 months ago
- surrogate fits for binary black-hole remnants☆10Dec 10, 2025Updated 2 months ago
- Package for data-driven and phenomenological gravitational waveform models☆12Feb 12, 2025Updated last year
- Next-generation pulsar-timing-array data analysis☆22Feb 27, 2026Updated last week
- Collection of layers that can perform arithmetic operations☆12Aug 17, 2021Updated 4 years ago
- A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations☆133Feb 20, 2026Updated 2 weeks ago
- Official Implementation of Memento☆21Nov 19, 2024Updated last year
- Example of GAMES (Gravitational-wave Amortized Metric Enhanced Sampler)☆12Oct 8, 2024Updated last year
- A jit-enhanced Python code to forecast the sensitivity of Pulsar Timing Array (PTA) configurations and assess constraints on Stochastic G…☆18Jan 20, 2026Updated last month
- A simple guitar tab viewer for NeoVim☆14May 21, 2020Updated 5 years ago
- An official repository for a VAE tutorial of Probabilistic Modelling and Reasoning - a University of Edinburgh master's course.☆10Jan 2, 2024Updated 2 years ago