https://github.com/cgarciae/nnx

Neural Networks for JAX
https://github.com/cgarciae/nnx

deep-learning jax neural-networks

Last synced: about 1 month ago
JSON representation

Neural Networks for JAX

• Host: GitHub
• URL: https://github.com/cgarciae/nnx
• Owner: cgarciae
• License: mit
• Archived: true
• Created: 2023-03-18T22:40:08.000Z (almost 2 years ago)
• Default Branch: main
• Last Pushed: 2024-09-24T13:00:11.000Z (3 months ago)
• Last Synced: 2024-11-16T09:37:08.665Z (about 2 months ago)
• Topics: deep-learning, jax, neural-networks
• Language: Python
• Homepage:
• Size: 1.43 MB
• Stars: 83
• Watchers: 5
• Forks: 2
• Open Issues: 5
• Metadata Files:
  • Readme: README.md
  • Funding: .github/FUNDING.yml
  • License: LICENSE

Awesome Lists containing this project

README

        
Moved to [flax/nnx](https://github.com/google/flax/tree/main/flax/nnx)