Issues
- 0
- 0
- 0
- 5
[Bug] elegy does not work with latest haiku version
#249 opened by murphyk - 0
CSVLogger iteration over a 0-d array
#248 opened by ScottAlexanderCameron - 0
Metrics ignore "on" keyword arg
#247 opened by ScottAlexanderCameron - 2
[Bug] Elegy crash on GPU
#246 opened by jiyuuchc - 0
- 5
- 0
Python 3.10 support
#244 opened by gabrielcnr - 0
LSTM
#242 opened by mwitiderrick - 0
- 0
- 0
AttributeError: module 'tensorflow.compiler.tf2xla.python.xla' has no attribute 'optimization_barrier'
#239 opened by mwitiderrick - 1
[Bug] Example is broken on GPU
#238 opened by CForgie - 1
Using Elegy with Equinox?
#235 opened by nalzok - 3
- 5
Pure functions in Elegy ?
#229 opened by pablo2909 - 1
[Feature Request] Gradient Accumulation
#225 opened by lkhphuc - 0
RFC: Elegy/Treex Ecosystem Next Versions
#230 opened by cgarciae - 0
Figure out how to test the Wandb callback on CI
#227 opened by cgarciae - 2
[Feature Request] Automatic mixed precision training.
#224 opened by lkhphuc - 2
- 6
[Bug] Problem with computing metrics
#216 opened by organic-chemistry - 0
Cleanup Documentation
#217 opened by cgarciae - 0
Random state within a Loss
#215 opened by cgarciae - 9
[Bug] Installing elegy is broken by jax v0.2.21
#189 opened by jfcrenshaw - 0
Add more metrics
#211 opened by anvelezec - 0
Add more Layers
#210 opened by cgarciae - 0
Develop R2 metric
#209 opened by anvelezec - 0
Develop IOU metric
#205 opened by anvelezec - 0
Develop cosine similarity metric
#206 opened by anvelezec - 0
Develop mean squared metric
#207 opened by anvelezec - 0
Develop root mean squared error metric
#208 opened by anvelezec - 0
Develop cohen kappa metric
#200 opened by anvelezec - 0
Develop precision and recall metrics
#201 opened by anvelezec - 0
Develop AUC metric
#202 opened by anvelezec - 0
Develop specificity metric
#203 opened by anvelezec - 0
Develop ROC metric
#204 opened by anvelezec - 0
Develop hamming distance metric
#198 opened by anvelezec - 0
Develop confusion matrix metric
#199 opened by anvelezec - 0
Community Enhancements
#197 opened by cgarciae - 4
[Bug] Low-level API Flax example is misleading.
#195 opened by lkhphuc - 0
Improve Summaries for all frameworks
#178 opened by cgarciae - 0
Define new summary API
#179 opened by cgarciae - 0
Implement new summary API for ElegyModule
#180 opened by cgarciae - 0
Implement new summary API for FlaxModule
#181 opened by cgarciae - 0
Implement new summary API for HaikuModule
#182 opened by cgarciae - 0
Implement new summary API for CallableModule
#183 opened by cgarciae - 8