Fast and memory efficient PyTorch implementation of the Perceiver with FlashAttention.
☆32Nov 4, 2024Updated last year
Alternatives and similar repositories for FlashPerceiver
Users that are interested in FlashPerceiver are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Falcon is a powerful, interpreted programming language.☆19Jan 22, 2023Updated 3 years ago
- Sampling techniques for Candle.☆21Apr 3, 2024Updated 2 years ago
- An autonomous robot, powered by AI.☆19May 3, 2022Updated 4 years ago
- Measuring the Signal to Noise Ratio in Language Model Evaluation☆30Aug 19, 2025Updated 10 months ago
- MIMEx: Intrinsic Rewards from Masked Input Modeling [NeurIPS 2023]☆16May 17, 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.
- ☆10Jun 27, 2024Updated 2 years ago
- Object-Centric-Representation Library (OCRL): This repo is to explore OCR on various downstream tasks from supervised learning tasks to R…☆12Feb 23, 2024Updated 2 years ago
- ☆16Jul 16, 2024Updated last year
- Official Implementation of `An Optimisation Framework for Unsupervised Environment Design` from RLC 2025☆17Nov 24, 2025Updated 7 months ago
- VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)☆29Jan 14, 2025Updated last year
- Pointax: PointMaze Environment for JAX☆28Oct 22, 2025Updated 8 months ago
- Official implementation of "Latent Action Learning Requires Supervision in the Presence of Distractors", ICML 2025☆37Jul 8, 2025Updated 11 months ago
- ☆22May 5, 2025Updated last year
- ☆24May 20, 2025Updated last year
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- ☆35Jun 21, 2024Updated 2 years ago
- Official codebase for "Sampling For Learnability", published at NeurIPS 2024☆23Oct 21, 2025Updated 8 months ago
- Scalable Opponent Shaping Experiments in JAX☆27Apr 13, 2024Updated 2 years ago
- ☆17Aug 20, 2025Updated 10 months ago
- Code of the Paper "Time-Efficient Reinforcement Learning with Stochastic Stateful Policies"☆25May 5, 2024Updated 2 years ago
- Code for the examples presented in the talk "Training a Llama in your backyard: fine-tuning very large models on consumer hardware" given…☆15Oct 16, 2023Updated 2 years ago
- ☆14Jul 17, 2024Updated last year
- ☆10Apr 6, 2024Updated 2 years ago
- Low rank adaptation (LoRA) for Candle.☆174Apr 18, 2025Updated last year
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Graph-Aware Attention for Adaptive Dynamics in Transformers☆70Jan 8, 2025Updated last year
- ☆15Dec 21, 2025Updated 6 months ago
- 📄Small Batch Size Training for Language Models☆82Mar 18, 2026Updated 3 months ago
- Accelerated replay buffers in JAX☆46Sep 17, 2022Updated 3 years ago
- Exploration into the Firefly algorithm in Pytorch☆41Feb 14, 2025Updated last year
- ☆12Sep 27, 2017Updated 8 years ago
- ☆12Jan 4, 2024Updated 2 years ago
- Dream Machine SDK for Python☆45May 13, 2026Updated last month
- A high-performance reinforcement learning library in jax specialized for robotic learning☆22Sep 4, 2023Updated 2 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.
- A rust wrapper for HIP☆13Jun 10, 2025Updated last year
- Data-Driven NetHack Tools: Datasets (30+) and recurrent-baselines (AWAC, BC, CQL, IQL, REM)☆43Aug 22, 2023Updated 2 years ago
- Minimalistic, hackable PyTorch implementation of SimSiam in ~400 lines. Achieves good performance on ImageNet with ResNet50. Features dis…☆22Nov 25, 2024Updated last year
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆24Oct 28, 2024Updated last year
- minimal Energy-based transformer☆44Dec 11, 2025Updated 6 months ago
- ☆16Sep 11, 2025Updated 9 months ago
- A dataloader, but for JAX☆20May 17, 2024Updated 2 years ago