https://github.com/cgarciae/nnx

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

deep-learning jax neural-networks

Last synced: 11 months 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 (over 3 years ago)
• Default Branch: main
• Last Pushed: 2024-09-24T13:00:11.000Z (almost 2 years ago)
• Last Synced: 2025-06-28T09:41:32.528Z (about 1 year ago)
• Topics: deep-learning, jax, neural-networks
• Language: Python
• Homepage:
• Size: 1.43 MB
• Stars: 84
• Watchers: 4
• 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)