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:
- ☆24Dec 11, 2024Updated last year
- ☆34Sep 10, 2024Updated last year
- Utilities for PyTorch distributed☆25Feb 27, 2025Updated last year
- A JAX implementation of the continuous time formulation of Consistency Models☆85Apr 7, 2023Updated 3 years ago
- Differentiable and gpu enabled fast wavelet transforms in JAX.☆48Jul 5, 2024Updated last year
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Unofficial JAX implementation of the SOAP optimizer (https://arxiv.org/abs/2409.11321)☆25Jan 9, 2026Updated 3 months ago
- Julia implementation of the Flash Attention algorithm☆19Sep 4, 2023Updated 2 years ago
- Implementation of the Gaussian Process Latent Variable Model.☆14Oct 28, 2022Updated 3 years ago
- Latent Diffusion Language Models☆70Sep 20, 2023Updated 2 years ago
- An unofficial implementation of the Fourier Neural Operator in Flax☆19May 25, 2024Updated last year
- A spoken version of the textual story cloze benchmark☆22Aug 6, 2023Updated 2 years ago
- ☆13Jun 3, 2024Updated last year
- Simple, blazing fast, transformer components.☆24Apr 1, 2023Updated 3 years ago
- ☆11Oct 15, 2025Updated 6 months 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.
- LoRA for arbitrary JAX models and functions☆144Feb 26, 2024Updated 2 years ago
- WIP☆95Aug 13, 2024Updated last year
- Focused on fast experimentation and simplicity☆80Dec 24, 2024Updated last year
- ☆10May 5, 2017Updated 8 years ago
- Offline RL experiments☆15Oct 1, 2022Updated 3 years ago
- Python tools☆14Oct 22, 2023Updated 2 years ago
- Personal website☆16Feb 20, 2026Updated 2 months ago
- 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
- 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.
- Deep neural models for core NLP tasks☆13Nov 9, 2017Updated 8 years ago
- ☆11Apr 25, 2025Updated last year
- CUDA implementation of autoregressive linear attention, with all the latest research findings☆46May 23, 2023Updated 2 years ago
- Code that accompanies the paper Guided Deep Kernel Learning☆10Jul 2, 2023Updated 2 years ago
- ☆12Jan 17, 2025Updated last year
- ☆93Jul 5, 2024Updated last year
- Repository for tutorial on Neural ODEs prepared for the UCL AI Society☆13Mar 7, 2021Updated 5 years ago
- ☆10Jun 13, 2022Updated 3 years ago
- ☆14Aug 15, 2024Updated last year
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Repo for the IDESSAI 2024 course on modeling audio with discrete tokens.☆13Sep 13, 2024Updated last year
- Efficient Conway's Game of Life implemented in Python using NumPy.☆14May 1, 2024Updated 2 years ago
- Code for ICLR 2024 paper "When should we prefer Decision Transformers for Offline Reinforcement Learning?"☆17Jan 31, 2024Updated 2 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 3 years ago
- ☆23Oct 15, 2022Updated 3 years ago
- Perceiver (transformer variant) implemented in JAX and Flax☆13Mar 29, 2021Updated 5 years ago