issues
search
alvarobartt
/
safejax
Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
https://alvarobartt.github.io/safejax/
MIT License
42
stars
5
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
bump versions for jax, jaxlib, objax, flax, chex (fixes #32)
#33
ihh
opened
1 year ago
6
Upgrade orbax deps?
#32
ihh
opened
1 year ago
4
Incompatibility issue with flax
#30
miguelamendez
closed
1 year ago
2
📝 Update `README.md` and `docs/`
#29
alvarobartt
closed
1 year ago
0
🐛 Fix issue when restoring a `VarCollection`
#28
alvarobartt
closed
1 year ago
0
🧪 Add assertions over `pytrees` with `chex`
#27
alvarobartt
closed
1 year ago
0
✨ Add `metadata` in `safejax.serialize`
#26
alvarobartt
closed
1 year ago
1
📝 Update `pull_request_template.md`
#25
alvarobartt
closed
1 year ago
0
Add `chex` assertions to unit tests
#24
alvarobartt
closed
1 year ago
1
🍱 Add support for `fsspec` file-systems
#23
alvarobartt
closed
1 year ago
0
Add `metadata` as a field in `serialize`
#22
alvarobartt
closed
1 year ago
1
♻️ Refactor on how `safejax` is structured
#21
alvarobartt
closed
1 year ago
0
Explore `safe_open` and see whether it's useful to integrate that too into `safejax`
#20
alvarobartt
closed
1 year ago
3
Add `fsspec` support to either upload or download `.safetensors` file from any fs
#19
alvarobartt
closed
1 year ago
0
Fix URL to `examples/` in `docs/usage.md`
#18
alvarobartt
closed
1 year ago
0
Move `load.py` and `save.py` to `core/` and define `partial` functions for `flax`, `haiku`, and `objax`?
#17
alvarobartt
closed
1 year ago
0
📝 Update `README.md` and `docs/usage.md`
#16
alvarobartt
closed
1 year ago
0
🔒 Lock dependencies with `pip-tools`
#15
alvarobartt
closed
1 year ago
0
📝 Add `mkdocs` and `mkdocs-material` documentation
#14
alvarobartt
closed
1 year ago
0
✏️ Add `DictionaryLike` type in `safejax.typing`
#13
alvarobartt
closed
1 year ago
0
Add unit tests for `objax` and `haiku` model params
#12
alvarobartt
closed
1 year ago
0
Define custom type in `safejax.typing` for model params
#11
alvarobartt
closed
1 year ago
0
✨ `safejax` support for `objax` model params
#10
alvarobartt
closed
1 year ago
0
Generate docs with `mkdocs` and `mkdocs-material` and publish to GitHub Pages
#9
alvarobartt
closed
1 year ago
0
Add `pip-tools` to lock the dependencies
#8
alvarobartt
closed
1 year ago
0
Explore integration with `objax`
#7
alvarobartt
closed
1 year ago
0
⚡️ Overall improvements, bug fixes, more unit tests, and `dm-haiku` compatibility tested
#6
alvarobartt
closed
1 year ago
0
✨ Add `benchmarks/resnet50.py`
#5
alvarobartt
closed
1 year ago
0
⬆️ Upgrade `safejax` to 0.1.1
#4
alvarobartt
closed
1 year ago
0
🧪 Add more unit test cases e.g. ResNet FrozenDict
#3
alvarobartt
closed
1 year ago
0
🚚 Rename `frozen_or_unfrozen_dict` to `params` in `serialize`
#2
alvarobartt
closed
1 year ago
0
✨ Add Flax-serialization with `safetensors` for `FrozenDict`
#1
alvarobartt
closed
1 year ago
0