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:
- Official implementation of "Token Perturbation Guidance for Diffusion Models" [NeurIPS 2025]☆18May 19, 2026Updated 3 months ago
- ☆33Sep 10, 2024Updated last year
- ☆24Dec 11, 2024Updated last year
- Utilities for PyTorch distributed☆26Feb 27, 2025Updated last year
- A JAX implementation of the continuous time formulation of Consistency Models☆83Apr 7, 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.
- Derivative-Informed Neural Operator: An Efficient Framework for High-Dimensional Parametric Derivative Learning☆20Jan 9, 2024Updated 2 years ago
- Implementation of LiteVAE☆19Feb 11, 2025Updated last year
- ☆16Dec 11, 2025Updated 8 months ago
- An unofficial implementation of the Fourier Neural Operator in Flax☆19Aug 12, 2026Updated 2 weeks ago
- LoRA for arbitrary JAX models and functions☆143Feb 26, 2024Updated 2 years ago
- Code to reproduce results from "Invertible generative models for inverse problems: mitigating representation error and dataset bias"☆21Jul 9, 2020Updated 6 years ago
- WIP☆96Aug 13, 2024Updated 2 years ago
- JAX Arrays for human consumption☆113Oct 23, 2025Updated 10 months ago
- Object-oriented Jax framework extending Equinox for scientific programming☆28May 7, 2026Updated 3 months 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.
- Focused on fast experimentation and simplicity☆77Dec 24, 2024Updated last year
- Offline RL experiments☆15Oct 1, 2022Updated 3 years ago
- Personal website☆16Jul 22, 2026Updated last month
- unofficial pytorch implementation of HiFi-GAN with fast MISR.☆15Mar 21, 2023Updated 3 years ago
- Official code for "On Calibrating Diffusion Probabilistic Models"☆30Feb 22, 2023Updated 3 years ago
- Deep neural models for core NLP tasks☆13Nov 9, 2017Updated 8 years ago
- The PyGpPhs package is designed for utilizing Gaussian Process for port-Hamiltonian system.☆11May 6, 2024Updated 2 years ago
- CUDA implementation of autoregressive linear attention, with all the latest research findings☆46May 23, 2023Updated 3 years ago
- Unofficial JAX implementation of the SOAP optimizer (https://arxiv.org/abs/2409.11321)☆29Jul 21, 2026Updated last month
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- ☆11Apr 25, 2025Updated last year
- Code that accompanies the paper Guided Deep Kernel Learning☆10Jul 2, 2023Updated 3 years ago
- Python 2D Navier-Stokes solver☆31Aug 4, 2025Updated last year
- ☆10Jun 13, 2022Updated 4 years ago
- We present cod-bench containing 12 operators and 10 datasets.☆11Jun 5, 2024Updated 2 years ago
- Repo for the IDESSAI 2024 course on modeling audio with discrete tokens.☆13Sep 13, 2024Updated last year
- ☆23Oct 15, 2022Updated 3 years ago
- The demo for "Discretization and Re-synthesis: an alternative method to solve the Cocktail Party Problem".☆12Oct 25, 2021Updated 4 years ago
- Perceiver (transformer variant) implemented in JAX and Flax☆13Mar 29, 2021Updated 5 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Differentiable Spline Approximations☆14Sep 3, 2023Updated 2 years ago
- Invertible Generative Flows☆85Apr 1, 2021Updated 5 years ago
- ☆24Oct 15, 2024Updated last year
- Code used in paper submitted to ICASSP 2019☆14Nov 22, 2018Updated 7 years ago
- ☆11Feb 8, 2023Updated 3 years ago
- An Arduino library for Multi-functional Energy Meter SDM120 Modbus☆10May 8, 2019Updated 7 years ago
- A repository with implementations of major papers on Gaussian Process regression models, implemented from scratch in Python, notably incl…☆14Nov 19, 2022Updated 3 years ago