issues
search
cgarciae
/
treex
A Pytree Module system for Deep Learning in JAX
https://cgarciae.github.io/treex/
MIT License
215
stars
17
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Fix LossAndLog error when there is no loss. (#75)
#79
jiyuuchc
closed
1 year ago
0
Instances of the same tx.Module have different tree_structure
#78
jiyuuchc
opened
1 year ago
3
Relax dependency restrictions
#77
cgarciae
closed
1 year ago
0
Recommended way to save/load tx.Modules?
#76
bhoov
opened
1 year ago
1
loss_and_log fails when there is no loss
#75
jiyuuchc
opened
1 year ago
0
Very small correction
#74
AdityaKane2001
opened
2 years ago
1
Take any version of cerifi
#73
jonringer
closed
2 years ago
1
Comparing Treex with Equinox
#72
nalzok
closed
2 years ago
5
Updates docs after refactor
#71
cgarciae
opened
2 years ago
0
Immutability Support
#70
cgarciae
opened
2 years ago
1
RFC: Elegy/Treex Ecosystem Next Versions
#69
cgarciae
opened
2 years ago
8
Latest treex is incompatible with the latest treeo
#68
samuela
closed
2 years ago
8
Try Fix CI
#67
cgarciae
closed
2 years ago
1
[WIP] Add Attention module
#66
lkhphuc
opened
2 years ago
0
Support python 3.7.0
#65
cgarciae
closed
2 years ago
1
Add conv_transpose
#64
lkhphuc
closed
2 years ago
1
Add create release workflow
#63
cgarciae
closed
2 years ago
1
Module.tabulate to return the Table
#62
soumik12345
opened
2 years ago
2
Release CI
#61
cgarciae
closed
2 years ago
0
Bumps `flax` to `0.4.0`
#60
ptigwe
closed
2 years ago
1
Call Module.__call__ during init by default
#59
cgarciae
opened
2 years ago
0
Add LayerNorm and GroupNorm
#58
lkhphuc
closed
2 years ago
2
Issue with case sensitivy file in docs/ ?
#57
lkhphuc
closed
2 years ago
3
Fix metric argument names
#56
cgarciae
closed
2 years ago
0
MeanSquaredError and MeanAbsoluteError Metrics use old argument names
#55
cgarciae
closed
2 years ago
0
Add Embed layer
#54
cgarciae
closed
2 years ago
0
Test import on CI
#53
cgarciae
closed
2 years ago
0
Update treeo
#52
cgarciae
closed
2 years ago
0
Fix documentation
#51
cgarciae
closed
2 years ago
0
Add MSE and tests
#50
jmarrietar
closed
2 years ago
1
Jit/Vmap-able DataLoader
#49
pharringtonp19
opened
2 years ago
2
Initial implementation of GRU layers
#48
ptigwe
closed
2 years ago
11
Add `trainable_parameters` for extracting non-frozen parameters
#47
ptigwe
closed
2 years ago
0
Community Resources
#46
cgarciae
opened
2 years ago
1
WIP: Implementation of Recurrent layers
#45
ptigwe
closed
2 years ago
2
Remove duplicate import of Haiku Module
#44
ptigwe
closed
2 years ago
1
Update select_topk to not use deprecated function
#43
ptigwe
closed
2 years ago
1
Fix typo
#42
BioGeek
closed
2 years ago
1
Add more metrics
#41
anvelezec
opened
2 years ago
1
Add more Layers
#40
cgarciae
opened
2 years ago
2
Develop R2 metric
#39
anvelezec
closed
2 years ago
0
Develop root mean squared error metric
#38
anvelezec
closed
2 years ago
0
Develop mean squared error metric
#37
anvelezec
closed
2 years ago
0
Develop cosine similarity metric
#36
anvelezec
closed
2 years ago
0
Develop IOU metric
#35
anvelezec
closed
2 years ago
0
Develop ROC metric
#34
anvelezec
closed
2 years ago
0
Develop specificity metric
#33
anvelezec
closed
2 years ago
0
Develop AUC metric
#32
anvelezec
closed
2 years ago
0
Develop precision and recall metric
#31
anvelezec
closed
2 years ago
0
Develop cohen kappa metric
#30
anvelezec
closed
2 years ago
0
Next