issues
search
instadeepai
/
flashbax
⚡ Flashbax: Accelerated Replay Buffers in JAX
https://instadeepai.github.io/flashbax/
Apache License 2.0
209
stars
10
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
prioritised_trajectory_buffer: allow adding max_length_time_axis
#40
garymm
closed
4 days ago
5
Replace functools.partial with jax.tree_util.Partial
#39
eadadi
opened
1 month ago
4
jax.tree_map -> jax.tree.map
#38
garymm
closed
1 month ago
2
Index trajectories at particular index from buffer
#37
shariqfz
opened
1 month ago
1
chore: use gh hosted runners for CI
#36
callumtilbury
closed
3 months ago
2
chore: Type ignore on max_length_time_axis
#35
callumtilbury
closed
3 months ago
1
feat: simple utility to get timestep count
#34
callumtilbury
closed
2 months ago
3
[Question] How to get the current buffer size from a buffer_state?
#33
ZaberKo
closed
2 months ago
11
fix: error when vault's rel_dir is a parent directory
#32
callumtilbury
closed
2 months ago
5
add is slow on GPU
#31
garymm
opened
3 months ago
5
feat: Mixed Experience Replay 🤝
#30
callumtilbury
closed
1 month ago
2
Needing help understanding trajectory buffer
#29
tobiaswuerth
closed
4 months ago
4
fix: linter for readme
#28
EdanToledo
closed
4 months ago
1
fix: be explicit about squeeze dim in prioritised sampling to avoid flattening (1,1) arrays
#27
callumtilbury
closed
4 months ago
0
JIT-compatible mini-batch creation without replacement for epoch-based training
#26
riiswa
closed
4 months ago
2
Update README.md
#25
EdanToledo
closed
5 months ago
0
Fix Jumanji environment spec access in examples
#24
rk1a
closed
6 months ago
1
What is the best buffer for my use-case of wanting to sample batches of independent episodes (either full or truncated, padded arrays)?
#23
Chulabhaya
closed
6 months ago
4
fix: improve the compression api of vaults
#22
callumtilbury
closed
7 months ago
1
chore: bump version to 0.1.1
#21
callumtilbury
closed
7 months ago
0
fix: incorrect current index.
#20
callumtilbury
closed
8 months ago
0
feat: add a prioritised item buffer
#19
EdanToledo
closed
4 months ago
4
feat: offer alternative backend read options
#18
callumtilbury
closed
8 months ago
3
Flashbax in-place updating of buffer-state compatibility with end-to-end jitted RL training style of PureJAXRL or JaxMARL?
#17
Chulabhaya
closed
7 months ago
14
feat: Improving the flexibility of reading vaults
#16
callumtilbury
closed
4 months ago
1
Support using vault as efficient buffer checkpointing mechanism for online RL
#15
vmichals
closed
8 months ago
4
feat: Vault Compression
#14
callumtilbury
closed
8 months ago
0
add: IQL Example
#13
JemmaLDaniel
closed
8 months ago
1
fix: mkdocs render issue of asterisk
#12
callumtilbury
closed
8 months ago
1
fix: release workflow
#11
EdanToledo
closed
9 months ago
0
fix: version in pyproject.toml
#10
callumtilbury
closed
9 months ago
0
chore: readme typo
#9
callumtilbury
closed
9 months ago
0
chore: push new fbx version to pypi
#8
jcformanek
closed
9 months ago
0
docs: buffer donation now works on cpu.
#7
callumtilbury
closed
10 months ago
0
Flat buffer add on CPU is inefficient
#6
maxencefaldor
closed
10 months ago
10
feat: vault
#5
callumtilbury
closed
9 months ago
0
Feat/item buffer
#4
EdanToledo
closed
10 months ago
1
Flat buffer seems to be unable to handle `truncated` correctly
#3
mantle2048
closed
11 months ago
2
Chore/update benchmark plots
#2
EdanToledo
closed
1 year ago
1
fix: clone repo and install example requirements
#1
EdanToledo
closed
1 year ago
0