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
- 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
- An autonomous robot, powered by AI.☆19May 3, 2022Updated 4 years ago
- MIMEx: Intrinsic Rewards from Masked Input Modeling [NeurIPS 2023]☆16May 17, 2023Updated 3 years ago
- ☆16Jul 16, 2024Updated 2 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- 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
- Generative method to design novel proteins using a diffusion model☆43Mar 16, 2023Updated 3 years ago
- ☆33Oct 22, 2024Updated last year
- [ICML 2026 GenBio Workshop] Official Implementation for "Harmonic Torsional Diffusion for Protein-Ligand Flexible Docking"☆15Jun 30, 2026Updated last month
- Measuring the Signal to Noise Ratio in Language Model Evaluation☆31Aug 19, 2025Updated 11 months ago
- ☆34Jun 21, 2024Updated 2 years ago
- Graph model execution API for Candle☆18Jul 27, 2025Updated last year
- ☆19Apr 22, 2024Updated 2 years ago
- ☆19Jul 4, 2025Updated last year
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Official Implementation of NeurIPS'23 Paper "Cross-Episodic Curriculum for Transformer Agents"☆32Oct 12, 2023Updated 2 years ago
- streaming deep reinforcement learning but 4x faster with jax!☆19Jan 4, 2026Updated 7 months ago
- ☆17Aug 20, 2025Updated 11 months ago
- Code of the Paper "Time-Efficient Reinforcement Learning with Stochastic Stateful Policies"☆25May 5, 2024Updated 2 years ago
- Official Implementation of `An Optimisation Framework for Unsupervised Environment Design` from RLC 2025☆17Nov 24, 2025Updated 8 months ago
- Scalable Opponent Shaping Experiments in JAX☆27Apr 13, 2024Updated 2 years ago
- Low rank adaptation (LoRA) for Candle.☆175Apr 18, 2025Updated last year
- ☆15Dec 21, 2025Updated 7 months ago
- ☆15Jul 17, 2024Updated 2 years ago
- 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.
- 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
- Graph-Aware Attention for Adaptive Dynamics in Transformers☆70Jan 8, 2025Updated last year
- Accelerated replay buffers in JAX☆46Sep 17, 2022Updated 3 years ago
- Exploration into the Firefly algorithm in Pytorch☆41Feb 14, 2025Updated last year
- 🔀 yet another mixture of experts☆23Jun 5, 2026Updated 2 months ago
- Vintix II: Decision Pre-Trained Transformer is a Scalable In-Context Reinforcement Learner - - = ICLR 2026☆16Apr 8, 2026Updated 4 months ago
- Data-Driven NetHack Tools: Datasets (30+) and recurrent-baselines (AWAC, BC, CQL, IQL, REM)☆43Aug 22, 2023Updated 2 years 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
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- ☆12Jan 4, 2024Updated 2 years ago
- ☆12Sep 27, 2017Updated 8 years ago
- A rust wrapper for HIP☆13Jun 10, 2025Updated last year
- A high-performance reinforcement learning library in jax specialized for robotic learning☆22Sep 4, 2023Updated 2 years ago
- A faster Arc.☆90Feb 8, 2024Updated 2 years ago
- ☆20Oct 5, 2025Updated 10 months ago
- ☆22May 5, 2025Updated last year