parallax

Module system

A module system for JAX that aims to simplify the creation and management of neural network models by providing an immutable tree-like structure for parameters.

GitHub
155 stars
6 watching
4 forks
Language: Python
last commit: over 5 years ago
Linked from 1 awesome list


Backlinks from these awesome lists:

Related projects:

Repository Description Stars
retraigo/la-classy A machine learning module for single layer perceptron models using Rust. 4
sebbadk/barion A compiled prompt system for various shells, providing a fast and simple way to display status information 1
marszall87/lambda-pure A fast and minimal ZSH prompt system with features like Git branch status and NodeJS version display. 121
ansimorph/oblong A customizable bash-inspired prompt that displays Git information and user details in a ZSH environment. 2
asem000/kernex A JAX-based stencil computation library for efficient numerical computations 67
pylons/horus A user registration and login system for the Pyramid Web Framework. 14
git-n-pissed/mprsa A MicroPython module for working with RSA keys 4
sonsongithub/numsw A Swift library for linear algebra and matrix operations with a playground-like notebook interface. 133
toni-sm/skrl A modular reinforcement learning library with support for various environments and frameworks 588
wintermi/zsh-starship A zsh plugin to customize and manage the Starship prompt 18
hszhao/pspnet A PyTorch implementation of a deep learning model for semantic image segmentation 1,598
charlax/professional-programming A collection of learning resources and documentation for software engineers to improve their skills in professional programming. 46,845
gpustack/gpustack Manages GPU clusters for running AI models 850
sohnryang/lambda-minimal-theme A simple zsh prompt theme that uses lambda to display directory information and virtual environment status 5
hendrixstring/android-triorm An Android-based ORM system for efficient data storage and querying using JVM serialization and minimal SQL wrapping. 19