Stainless neural networks in JAX
☆37Jun 13, 2026Updated 4 months ago
Alternatives and similar repositories for inox
Users that are interested in inox are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆14Aug 8, 2023Updated 3 years ago
- Documentation and guidelines for the Alan GPU cluster at the University of Liège.☆22Jul 19, 2023Updated 3 years ago
- Unleash the power of scheduling☆40Oct 31, 2026Updated last week
- A Jax wrapper for cudaKDTree☆11Sep 26, 2025Updated last year
- GATSBI: Generative Adversarial Training for Simulation-Based Inference☆20Jul 13, 2023Updated 3 years 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.
- Exploring how ChatGPT can be used to accelerate research in cosmology.☆13Dec 12, 2022Updated 3 years ago
- A Jax package to perform Neural Density Estimation from cosmological probes.☆13Jun 19, 2026Updated 4 months ago
- A Python toolkit for (simulation-based) inference and the mechanization of science.☆55Apr 15, 2022Updated 4 years ago
- A minimal poster template in Typst.☆28Oct 24, 2025Updated last year
- Differentiable and accelerated wavelet transform on the sphere with JAX☆26May 27, 2025Updated last year
- Likelihood-free AMortized Posterior Estimation with PyTorch☆137Aug 21, 2024Updated 2 years ago
- Euclid Visible Instrument Python package. Includes a simulator and various analysis codes.☆11Mar 25, 2015Updated 11 years ago
- Meeting repo for likelihood free inference meeting☆14Oct 6, 2022Updated 4 years ago
- An experiment: emcee implemented in JAX☆28Jul 1, 2022Updated 4 years 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.
- A TensorFlow reimplementation of GalSim☆10Mar 8, 2022Updated 4 years ago
- ☆12Nov 5, 2025Updated last year
- Jaxpr Visualisation Tool☆38Dec 22, 2024Updated last year
- Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.☆21Mar 10, 2024Updated 2 years ago
- Code for Gaussian Score Matching Variational Inference☆34Feb 23, 2025Updated last year
- Semantic alignment of astronomical data with natural language using multi-modal models. (Jax) Code associated with https://arxiv.org/abs/…☆18Oct 18, 2024Updated 2 years ago
- Code for the paper "Towards Reliable Simulation-Based Inference with Balanced Neural Ratio Estimation".☆15Nov 14, 2022Updated 3 years ago
- Diffusion models in PyTorch☆134Aug 28, 2026Updated 2 months ago
- Community-sourced list of papers and resources on neural simulation-based inference.☆169Oct 26, 2026Updated last week
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Generative models with JAX.☆22Feb 21, 2026Updated 8 months ago
- Differentiable and gpu enabled fast wavelet transforms in JAX.☆48Jul 5, 2024Updated 2 years ago
- Benchmarking field-level cosmological inference from galaxy surveys.☆16Jul 17, 2025Updated last year
- Materials for DATS0001 Foundations of Data Science, ULiège☆49Updated this week
- Repository with the code of "HNPE: Leveraging Global Parameters for Neural Posterior Estimation"☆14Mar 18, 2024Updated 2 years ago
- A functional training loops library for JAX☆87Feb 13, 2024Updated 2 years ago
- Shape Measurement Pipeline☆18Updated this week
- The galaxy survey codex☆17Oct 1, 2026Updated last month
- ✨💯Gaussian Mixture Models in JAX 💯✨☆26Aug 7, 2026Updated 3 months ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- simple simulations for testing weak lensing shear measurement☆14Apr 24, 2026Updated 6 months ago
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆51Dec 9, 2024Updated last year
- Pytorch-like dataloaders for JAX.☆105Dec 16, 2025Updated 10 months ago
- A simple implementation of Hamiltonian Monte Carlo in JAX.☆20Feb 8, 2024Updated 2 years ago
- Offical code for "L-C2ST: Local Validation Diagnostics for Posterior Approximations in Simulation-Based Inference", NeurIPS 2023☆11Jun 10, 2024Updated 2 years ago
- Official implementation of Learning Diffusion Priors from Observations by Expectation Maximization☆38Mar 19, 2026Updated 7 months ago
- Galactic and Gravitational Dynamics in Python (+ GPU and autodiff)☆53Updated this week