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
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 3 years ago
- Falcon is a powerful, interpreted programming language.☆17Jan 22, 2023Updated 3 years ago
- Measuring the Signal to Noise Ratio in Language Model Evaluation☆29Aug 19, 2025Updated 7 months ago
- MIMEx: Intrinsic Rewards from Masked Input Modeling [NeurIPS 2023]☆16May 17, 2023Updated 2 years ago
- ☆10Jun 27, 2024Updated last year
- ☆16Jul 16, 2024Updated last year
- Pointax: PointMaze Environment for JAX☆26Oct 22, 2025Updated 4 months ago
- Reinforcement Learning inside a 3D soccer simulation☆37Sep 15, 2024Updated last year
- VC-FB and MC-FB algorithms from "Zero-Shot Reinforcement Learning from Low Quality Data" (NeurIPS 2024)☆27Jan 14, 2025Updated last year
- [ICML 2025 GenBio Workshop] Official Implementation for "Electrostatics from Laplacian Eigenbasis for Neural Network Interatomic Potentia…☆17Jun 12, 2025Updated 9 months ago
- ☆33Oct 22, 2024Updated last year
- Official implementation of "Latent Action Learning Requires Supervision in the Presence of Distractors", ICML 2025☆33Jul 8, 2025Updated 8 months ago
- ☆22May 5, 2025Updated 10 months ago
- Graph model execution API for Candle☆17Jul 27, 2025Updated 7 months ago
- ☆32Jun 21, 2024Updated last year
- ☆19Apr 22, 2024Updated last year
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆31Oct 12, 2023Updated 2 years ago
- ☆18Aug 20, 2025Updated 7 months ago
- Code of the Paper "Time-Efficient Reinforcement Learning with Stochastic Stateful Policies"☆25May 5, 2024Updated last year
- ☆18Jul 4, 2025Updated 8 months 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
- ☆11Apr 6, 2024Updated last year
- ☆14Dec 21, 2025Updated 3 months ago
- High-performance JAX-powered simulator for robotic navigation in 2D mazes, optimized for Quality-Diversity algorithm research and benchma…☆20Jun 19, 2025Updated 9 months ago
- Accelerated replay buffers in JAX☆46Sep 17, 2022Updated 3 years ago
- ☆12Sep 27, 2017Updated 8 years ago
- A high-performance reinforcement learning library in jax specialized for robotic learning☆22Sep 4, 2023Updated 2 years ago
- Dream Machine SDK for Python☆43Updated this week
- A rust wrapper for HIP☆12Jun 10, 2025Updated 9 months ago
- 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…☆21Nov 25, 2024Updated last year
- ☆19Mar 1, 2023Updated 3 years ago
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆23Oct 28, 2024Updated last year
- minimal Energy-based transformer☆43Dec 11, 2025Updated 3 months ago
- A dataloader, but for JAX☆20May 17, 2024Updated last year
- Code for the Cartoon Set webpage.☆15May 21, 2020Updated 5 years ago
- Rust Workspace Bootstrapper☆18Oct 5, 2025Updated 5 months ago
- Code for the note "NF4 Isn't Information Theoretically Optimal (and that's Good)☆21Jun 22, 2023Updated 2 years ago