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]☆17May 19, 2026Updated 2 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
- 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.
- Derivative-Informed Neural Operator: An Efficient Framework for High-Dimensional Parametric Derivative Learning☆19Jan 9, 2024Updated 2 years ago
- Differentiable and gpu enabled fast wavelet transforms in JAX.☆48Jul 5, 2024Updated 2 years ago
- Implementation of LiteVAE☆19Feb 11, 2025Updated last year
- Latent Diffusion Language Models☆72Sep 20, 2023Updated 2 years ago
- ☆16Dec 11, 2025Updated 8 months ago
- An unofficial implementation of the Fourier Neural Operator in Flax☆19May 25, 2024Updated 2 years ago
- ☆13Jun 3, 2024Updated 2 years 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
- Open source password manager - Proton Pass • AdSecurely store, share, and autofill your credentials with Proton Pass, the end-to-end encrypted password manager trusted by millions.
- WIP☆96Aug 13, 2024Updated last year
- JAX Arrays for human consumption☆113Oct 23, 2025Updated 9 months ago
- Object-oriented Jax framework extending Equinox for scientific programming☆27May 7, 2026Updated 3 months ago
- Focused on fast experimentation and simplicity☆77Dec 24, 2024Updated last year
- Un-*** 50 billions multimodality dataset☆24Sep 14, 2022Updated 3 years ago
- Offline RL experiments☆15Oct 1, 2022Updated 3 years ago
- ☆10May 5, 2017Updated 9 years ago
- Personal website☆16Jul 22, 2026Updated 2 weeks ago
- unofficial pytorch implementation of HiFi-GAN with fast MISR.☆15Mar 21, 2023Updated 3 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- 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
- A spoken version of the textual story cloze benchmark☆22Aug 6, 2023Updated 3 years ago
- ☆12Jan 17, 2025Updated last year
- Python 2D Navier-Stokes solver☆31Aug 4, 2025Updated last year
- ☆93Jul 5, 2024Updated 2 years ago
- This repository will contain code for the paper "CLIP meets GamePhysics: Towards bug identification in gameplay videos using zero-shot tr…☆25Dec 23, 2023Updated 2 years ago
- We present cod-bench containing 12 operators and 10 datasets.☆11Jun 5, 2024Updated 2 years ago
- Repository for tutorial on Neural ODEs prepared for the UCL AI Society☆13Mar 7, 2021Updated 5 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Repo for the IDESSAI 2024 course on modeling audio with discrete tokens.☆13Sep 13, 2024Updated last year
- Code for ICLR 2024 paper "When should we prefer Decision Transformers for Offline Reinforcement Learning?"☆17Jan 31, 2024Updated 2 years ago
- ☆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
- [ICML 2022] Official implementation of "Score-Guided Intermediate Layer Optimization: Fast Langevin Mixing for Inverse Problems".☆12Jul 19, 2022Updated 4 years ago
- Perceiver (transformer variant) implemented in JAX and Flax☆13Mar 29, 2021Updated 5 years ago
- ☆27Dec 15, 2022Updated 3 years ago