Lightning-Universe / lightning-diffusion_component
☆26Updated last year
Related projects: ⓘ
- ☆38Updated this week
- Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing☆47Updated 2 years ago
- Contrastive Language-Image Pretraining☆37Updated 2 months ago
- Load any clip model with a standardized interface☆21Updated 4 months ago
- A scalable implementation of diffusion and flow-matching with XGBoost models, applied to calorimeter data.☆13Updated 3 weeks ago
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️☆52Updated last year
- ☆23Updated this week
- Implementation of Metaformer, but in an autoregressive manner☆22Updated 2 years ago
- Implementation of TableFormer, Robust Transformer Modeling for Table-Text Encoding, in Pytorch☆35Updated 2 years ago
- ☆40Updated this week
- An open source implementation of CLIP.☆32Updated last year
- A GPT, made only of MLPs, in Jax☆55Updated 3 years ago
- ☆24Updated this week
- FID computation in Jax/Flax.☆23Updated 2 months ago
- Implementation of a Transformer using ReLA (Rectified Linear Attention) from https://arxiv.org/abs/2104.07012☆49Updated 2 years ago
- ☆21Updated this week
- A dashboard for exploring timm learning rate schedulers☆18Updated last year
- Latent Diffusion Language Models☆66Updated last year
- Implementation of a holodeck, written in Pytorch☆17Updated 10 months ago
- A convolution-free, transformer-only version of the CycleGAN framework☆32Updated 2 years ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆74Updated 2 years ago
- JAX implementation ViT-VQGAN☆77Updated last year
- Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.☆22Updated last year
- Describe the format of image/text datasets☆11Updated 2 years ago
- Implementation of Tranception, an attention network, paired with retrieval, that is SOTA for protein fitness prediction☆31Updated 2 years ago
- Another attempt at a long-context / efficient transformer by me☆37Updated 2 years ago
- A python library for highly configurable transformers - easing model architecture search and experimentation.☆50Updated 2 years ago
- AdaCat☆49Updated 2 years ago
- A JAX nn library☆20Updated 6 months ago
- Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'☆38Updated 2 years ago