Giters
cgarciae
/
nnx
Neural Networks for JAX
Geek Repo:
Geek Repo
Github PK Tool:
Github PK Tool
Stargazers:
79
Watchers:
5
Issues:
7
Forks:
2
cgarciae/nnx Issues
Optimal way to handle Variable metadata
Updated
a year ago
Comments count
2
Move logic from RngStream to Context
Closed
a year ago
Don’t use id() as hash
Closed
a year ago
Use a single string instead of tuple for State keys
Closed
a year ago
Remove utils.py
Closed
a year ago
Test sharing lifted Modules
Updated
a year ago
Pedagogical implementation of jit/grad
Updated
a year ago
Comments count
1