Fast and memory efficient PyTorch implementation of the Perceiver with FlashAttention.
☆33Nov 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:
- 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
- MIMEx: Intrinsic Rewards from Masked Input Modeling [NeurIPS 2023]☆16May 17, 2023Updated 3 years ago
- ☆16Jul 16, 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
- ☆10Jun 27, 2024Updated 2 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Reinforcement Learning inside a 3D soccer simulation☆35Sep 15, 2024Updated 2 years ago
- ☆34Oct 22, 2024Updated last year
- [ICML 2026 GenBio Workshop] Official Implementation for "Harmonic Torsional Diffusion for Protein-Ligand Flexible Docking"☆17Jun 30, 2026Updated 3 months ago
- Measuring the Signal to Noise Ratio in Language Model Evaluation☆32Aug 19, 2025Updated last year
- ☆34Jun 21, 2024Updated 2 years ago
- ☆19Jul 4, 2025Updated last year
- streaming deep reinforcement learning but 4x faster with jax!☆19Jan 4, 2026Updated 9 months ago
- ☆17Aug 20, 2025Updated last year
- Code of the Paper "Time-Efficient Reinforcement Learning with Stochastic Stateful Policies"☆25May 5, 2024Updated 2 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Scalable Opponent Shaping Experiments in JAX☆27Apr 13, 2024Updated 2 years ago
- A deep learning-powered visual navigation engine to enables autonomous navigation of pocket-size quadrotor - running on PULP☆13Oct 30, 2024Updated last year
- ☆15Jul 17, 2024Updated 2 years ago
- Accelerated replay buffers in JAX☆47Sep 17, 2022Updated 4 years ago
- Exploration into the Firefly algorithm in Pytorch☆41Feb 14, 2025Updated last year
- 🔀 yet another mixture of experts☆23Jun 5, 2026Updated 4 months ago
- Vintix II: Decision Pre-Trained Transformer is a Scalable In-Context Reinforcement Learner - - = ICLR 2026☆17Apr 8, 2026Updated 6 months ago
- A scalable benchmark for state representation learning in visual reinforcement learning.☆17Jun 23, 2025Updated last year
- High-performance JAX-powered simulator for robotic navigation in 2D mazes, optimized for Quality-Diversity algorithm research and benchma…☆21Jun 19, 2025Updated last year
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- A high-performance reinforcement learning library in jax specialized for robotic learning☆22Sep 4, 2023Updated 3 years ago
- ☆23May 5, 2025Updated last year
- ☆19Mar 1, 2023Updated 3 years ago
- [ICML 2025 GenBio Workshop] Official Implementation for "Electrostatics from Laplacian Eigenbasis for Neural Network Interatomic Potentia…☆18Jun 12, 2025Updated last year
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆24Oct 28, 2024Updated last year
- Code for the note "NF4 Isn't Information Theoretically Optimal (and that's Good)☆20Jun 22, 2023Updated 3 years ago
- Code for the Cartoon Set webpage.☆15May 21, 2020Updated 6 years ago
- Convert 3D Human Pose to VMD file☆14Apr 21, 2019Updated 7 years ago
- Official codebase for "Sampling For Learnability", published at NeurIPS 2024☆25Oct 21, 2025Updated 11 months ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- High quality implementations of imitation and inverse reinforcement learning algorithms☆26Aug 19, 2025Updated last year
- Gym env for Slay the Spire☆19Dec 31, 2024Updated last year
- Official implementation of Adaptive Feature Transfer (AFT)☆25Jun 12, 2024Updated 2 years ago
- Multi-agent simulator in Jax for research and teaching in AI & ALife☆33Apr 11, 2026Updated 5 months ago
- Official Pytorch Implementation of "Zero-Shot Off-Policy Learning" (ICML 2026)☆25Feb 16, 2026Updated 7 months ago
- Learning Robust Dynamics Through Variational Sparse Gating☆21Oct 19, 2022Updated 3 years ago
- A dataloader, but for JAX☆20May 17, 2024Updated 2 years ago