Fast and memory efficient PyTorch implementation of the Perceiver with FlashAttention.
☆31Nov 4, 2024Updated last year
Alternatives and similar repositories for FlashPerceiver
Users that are interested in FlashPerceiver are comparing it to the libraries listed below
Sorting:
- The DynaGen model allows the prediction of dynamical field data based on microstructure input. The model is exemplified to dynamic fractu…☆12Mar 17, 2023Updated 2 years ago
- Falcon is a powerful, interpreted programming language.☆16Jan 22, 2023Updated 3 years ago
- PyTorch implementation for all methods and environments in the paper "MIMEx: Intrinsic Rewards from Masked Input Modeling"☆16May 17, 2023Updated 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
- [ICML 2025 GenBio Workshop] Official Implementation for "Electrostatics from Laplacian Eigenbasis for Neural Network Interatomic Potentia…☆17Jun 12, 2025Updated 8 months ago
- Pointax: PointMaze Environment for JAX☆26Oct 22, 2025Updated 4 months ago
- ☆16Jul 16, 2024Updated last year
- Sampling techniques for Candle.☆19Apr 3, 2024Updated last year
- ☆31Jun 21, 2024Updated last year
- Minimalistic, hackable PyTorch implementation of SimSiam in ~400 lines. Achieves good performance on ImageNet with ResNet50. Features dis…☆21Nov 25, 2024Updated last year
- VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)☆25Jan 14, 2025Updated last year
- Official codebase for "Sampling For Learnability", published at NeurIPS 2024☆20Oct 21, 2025Updated 4 months ago
- ☆18Jul 4, 2025Updated 7 months ago
- Official implementation of "Latent Action Learning Requires Supervision in the Presence of Distractors", ICML 2025☆33Jul 8, 2025Updated 7 months ago
- ☆18Aug 20, 2025Updated 6 months ago
- ☆22May 5, 2025Updated 9 months ago
- ☆19Apr 22, 2024Updated last year
- Accelerated replay buffers in JAX☆46Sep 17, 2022Updated 3 years ago
- Data-Driven NetHack Tools: Datasets (30+) and recurrent-baselines (AWAC, BC, CQL, IQL, REM)☆43Aug 22, 2023Updated 2 years ago
- High-performance JAX-powered simulator for robotic navigation in 2D mazes, optimized for Quality-Diversity algorithm research and benchma…☆20Jun 19, 2025Updated 8 months ago
- Official implementation of Adaptive Feature Transfer (AFT)☆23Jun 12, 2024Updated last year
- 📄Small Batch Size Training for Language Models☆80Oct 4, 2025Updated 4 months ago
- Learning Robust Dynamics Through Variational Sparse Gating☆20Oct 19, 2022Updated 3 years ago
- A dataloader, but for JAX☆20May 17, 2024Updated last year
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆23Oct 28, 2024Updated last year
- Accompanying Code for "Flipping Coins to Estimate Pseudocounts for Exploration in Reinforcement Learning", ICML 2023☆22Dec 29, 2023Updated 2 years ago
- A high-performance reinforcement learning library in jax specialized for robotic learning☆22Sep 4, 2023Updated 2 years ago
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆23Feb 15, 2025Updated last year
- minimal Energy-based transformer☆43Dec 11, 2025Updated 2 months ago
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆31Oct 12, 2023Updated 2 years ago
- Multi-agent simulator in Jax for research and teaching in AI & ALife☆31Updated this week
- [ICLR 2024] Closing the Gap between TD Learning and Supervised Learning - A Generalisation Point of View.☆23Apr 19, 2024Updated last year
- ☆24Aug 9, 2024Updated last year
- Code for Powderworld: A Platform for Understanding Generalization via Rich Task Distributions☆73Aug 31, 2024Updated last year
- Source code for the paper "Policy Architectures for Compositional Generalization in Control"☆30May 19, 2022Updated 3 years ago
- Minimal RLHF implementation built on top of minGPT.☆32Jul 4, 2024Updated last year
- When Do Transformers Shine in RL? Decoupling Memory from Credit Assignment, NeurIPS 2023 (oral)☆68Jan 18, 2024Updated 2 years ago
- Official codebase for "The Generalization Gap in Offline Reinforcement Learning" accepted to ICLR 2024☆28Feb 20, 2026Updated last week
- Official PyTorch implementation of "GuidedQuant: Large Language Model Quantization via Exploiting End Loss Guidance" (ICML 2025)☆50Jul 6, 2025Updated 7 months ago