vvvm23 / TchAIkovsky
Using JAX to generate piano music as MIDI
☆37Updated 9 months ago
Related projects: ⓘ
- Implementation of Perceiver AR, Deepmind's new long-context attention network based on Perceiver architecture, in Pytorch☆86Updated last year
- A State-Space Model with Rational Transfer Function Representation.☆61Updated 4 months ago
- Latent Diffusion Language Models☆66Updated last year
- Utilities for PyTorch distributed☆23Updated 11 months ago
- Collection of autoregressive model implementation☆62Updated 2 weeks ago
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆82Updated 3 weeks ago
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆45Updated last month
- ☆23Updated this week
- FID computation in Jax/Flax.☆23Updated 2 months ago
- The demo page of UniAudio☆34Updated 7 months ago
- Diffusion models in PyTorch☆81Updated last week
- Examples of apps built with Nendo, the AI Audio Tool Suite☆55Updated 6 months ago
- Implementation of a Light Recurrent Unit in Pytorch☆43Updated 3 weeks ago
- A JAX implementation of the continuous time formulation of Consistency Models☆83Updated last year
- ☆18Updated last month
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆29Updated 3 weeks ago
- Audiocraft is a library for audio processing and generation with deep learning. It features the state-of-the-art EnCodec audio compressor…☆51Updated 5 months ago
- Educational implementation of the Discrete Flow Matching paper☆24Updated 3 weeks ago
- Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto☆53Updated 4 months ago
- Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"☆56Updated 10 months ago
- Framework for writing deep learning training loops. Lightweight, and retaining full freedom to design as you see fits. It handles checkpo…☆101Updated 6 months ago
- An implementation of the Llama architecture, to instruct and delight☆21Updated last month
- ☆38Updated 2 months ago
- Train vision models using JAX and 🤗 transformers☆91Updated last week
- Implementation of GateLoop Transformer in Pytorch and Jax☆86Updated 3 months ago
- Clean RL implementation using MLX☆26Updated 6 months ago
- Offical code for the CVPR 2024 Paper: Separating the "Chirp" from the "Chat": Self-supervised Visual Grounding of Sound and Language☆53Updated 3 months ago
- ☆11Updated this week
- Explorations into the recently proposed Taylor Series Linear Attention☆85Updated last month
- RWKV model implementation☆38Updated last year