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
Develop confusion matrix metric
#29
anvelezec
closed
2 years ago
0
Develop hamming distance
#28
anvelezec
closed
2 years ago
0
Fix import error when Haiku is not available
#27
cgarciae
closed
2 years ago
0
[WIP] Multi head attention
#26
cgarciae
closed
2 years ago
1
Next key axis name
#25
cgarciae
closed
2 years ago
0
Update treeo
#24
cgarciae
closed
2 years ago
0
Adds HaikuModule
#23
cgarciae
closed
2 years ago
0
Update treeo
#22
cgarciae
closed
2 years ago
0
Improve docs
#21
cgarciae
closed
2 years ago
0
Update User Guide
#20
cgarciae
opened
2 years ago
0
Removes double assignment in nn.Lambda
#19
ptigwe
closed
2 years ago
1
Update readme + version
#18
cgarciae
closed
2 years ago
0
Feature/metrics
#17
anvelezec
closed
2 years ago
3
Shape Inference + Module Hooks (compact)
#16
cgarciae
closed
2 years ago
0
Lazy layer / Shape inference?
#15
lkhphuc
closed
2 years ago
1
Add losses
#14
cgarciae
closed
2 years ago
0
More Metrics
#13
cgarciae
closed
2 years ago
0
[WIP] MultiHeadAttention
#12
cgarciae
closed
2 years ago
0
0.5.0
#11
cgarciae
closed
2 years ago
0
Fix assert key jnp.ndarray error
#10
lkhphuc
closed
2 years ago
0
Module init fails at wrong key type
#9
kimbochen
closed
2 years ago
4
How to deal with shared submodules?
#8
avital
opened
2 years ago
8
xor logistic regression
#7
msaroufim
closed
3 years ago
3
Don't allow TreeParts inside generics
#6
cgarciae
closed
3 years ago
1
0.4.0
#5
cgarciae
closed
2 years ago
0
Refactor
#4
cgarciae
closed
3 years ago
0
Refactor Sequential
#3
cgarciae
closed
3 years ago
0
[RFC] Proposal to readme
#2
Davidnet
closed
3 years ago
1
RFC Proofreading and Grammar.
#1
Davidnet
closed
3 years ago
0
Previous