guillaumeBellec / multitask
☆22Updated 4 months ago
Related projects: ⓘ
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆87Updated 3 weeks ago
- Jax/Flax implementation of Variational-DiffWave.☆40Updated 2 years ago
- My attempts at applying Soundstream design on learned tokenization of text and then applying hierarchical attention to text generation☆76Updated last year
- ☆163Updated last year
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆74Updated 7 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
- A home for audio ML in JAX. Has common features, learnable frontends, pretrained supervised and self-supervised models.☆62Updated 2 years ago
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆123Updated 2 years ago
- Residual Quantization with Implicit Neural Codebooks☆44Updated last month
- Inspired by "Neural Networks Fail to Learn Periodic Functions and How to Fix It"☆56Updated 4 months ago
- Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne…☆115Updated last year
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆119Updated 11 months ago
- ☆61Updated last month
- Sequence Modeling with Structured State Spaces☆60Updated 2 years ago
- Implementation of Perceiver AR, Deepmind's new long-context attention network based on Perceiver architecture, in Pytorch☆86Updated last year
- Easy Hypernetworks in Pytorch and Jax☆94Updated last year
- Implementation of fused cosine similarity attention in the same style as Flash Attention☆204Updated last year
- Implementation of DiffWave and SaShiMi audio generation models☆112Updated last year
- A toolbox that provides hackable building blocks for generic 1D/2D/3D UNets, in PyTorch.☆77Updated last year
- ☆43Updated 2 years ago
- Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto☆53Updated 4 months ago
- ☆29Updated 8 months ago
- Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch☆94Updated last year
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆74Updated 2 years ago
- FID computation in Jax/Flax.☆23Updated 2 months ago
- A State-Space Model with Rational Transfer Function Representation.☆61Updated 4 months ago
- Repository for the "Gotta Go Fast When Generating Data with Score-Based Models" paper☆101Updated 2 years ago
- A tool for assignment to a slice in TensorFlow☆19Updated 3 years ago
- Why Do We Need Weight Decay in Modern Deep Learning? [arXiv, Oct 2023]☆41Updated 11 months ago
- Implementation of a Light Recurrent Unit in Pytorch☆43Updated 3 weeks ago