The 2D discrete wavelet transform for JAX
☆45Feb 28, 2023Updated 3 years ago
Alternatives and similar repositories for jax-wavelets
Users that are interested in jax-wavelets are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- A JAX implementation of the continuous time formulation of Consistency Models☆83Apr 7, 2023Updated 3 years ago
- Differentiable and gpu enabled fast wavelet transforms in JAX.☆48Jul 5, 2024Updated 2 years ago
- A Python library that integrates the Curvelet transform into differentiable programming pipelines, such as PyTorch and JAX.☆12Jun 22, 2023Updated 3 years ago
- Example code to speed up GP inference with gradients for high-dimensional inputs.☆14Jul 12, 2021Updated 5 years ago
- Score-based Diffusion models in JAX.☆19Dec 29, 2025Updated 10 months ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- ☆33Sep 10, 2024Updated 2 years ago
- DISCnetMachineLearningCourse☆10Jul 2, 2025Updated last year
- Derivative-Informed Neural Operator: An Efficient Framework for High-Dimensional Parametric Derivative Learning☆21Jan 9, 2024Updated 2 years ago
- Utilities for PyTorch distributed☆26Feb 27, 2025Updated last year
- Code to reproduce results from "Invertible generative models for inverse problems: mitigating representation error and dataset bias"☆22Jul 9, 2020Updated 6 years ago
- Differentiable and accelerated wavelet transform on the sphere with JAX☆26May 27, 2025Updated last year
- An unofficial implementation of the Fourier Neural Operator in Flax☆19Aug 12, 2026Updated 2 months ago
- Binding affinity prediction for drug discovery☆10Aug 28, 2025Updated last year
- ☆10May 5, 2017Updated 9 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Un-*** 50 billions multimodality dataset☆24Sep 14, 2022Updated 4 years ago
- Object-oriented Jax framework extending Equinox for scientific programming☆28Sep 23, 2026Updated last month
- Joint X-ray and SZ data fitting code for galaxy clusters in Python☆11Nov 24, 2021Updated 4 years ago
- An example of DyNet autobatching for the NIPS "how to code a paper" workshop☆12Dec 9, 2017Updated 8 years ago
- Latent Diffusion Language Models☆71Sep 20, 2023Updated 3 years ago
- ☆28Jan 17, 2023Updated 3 years ago
- ☆23May 6, 2020Updated 6 years ago
- AMI-GO brrrrrrr☆26Updated this week
- ☆11Oct 13, 2026Updated 3 weeks ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- Tutorials for the U Toronto Foundation Models for Science Workshop☆12Nov 10, 2025Updated 11 months ago
- homegrown reprojection algorithm using torch☆14Feb 20, 2026Updated 8 months ago
- Kernel Localized Linear Regression (KLLR)☆12Jan 31, 2024Updated 2 years ago
- Classic MCTS example with mctx☆25May 25, 2023Updated 3 years ago
- Code for the paper High Fidelity Image Synthesis With Deep VAEs In Latent Space.☆29Oct 25, 2023Updated 3 years ago
- version tracking for the swift analysis in support of the planck clusters☆10Jan 6, 2021Updated 5 years ago
- IRSA Tutorials☆17Updated this week
- The PyGpPhs package is designed for utilizing Gaussian Process for port-Hamiltonian system.☆11May 6, 2024Updated 2 years ago
- Official code for "On Calibrating Diffusion Probabilistic Models"☆30Feb 22, 2023Updated 3 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- ☆14Apr 22, 2022Updated 4 years ago
- Python tools☆14Oct 22, 2023Updated 3 years ago
- A small syntax highlighting JAR for Tcl in IntelliJ products☆15Jan 30, 2017Updated 9 years ago
- 🤓 A tutorial on the Discretized Logistic Mixture Likelihood (DLML)☆12Aug 10, 2024Updated 2 years ago
- Experiments for "D. Davis, D. Drusvyatskiy, and L. Jiang. Gradient descent with adaptive stepsize converges (nearly) linearly under fourt…☆10May 2, 2025Updated last year
- Code for LOFAR catalogue creation☆11Sep 6, 2026Updated last month
- ☆11Apr 25, 2025Updated last year