watcl-lab / neural_networks_and_computationLinks
Computational abilities and efficiency of neural networks
β57Updated 2 months ago
Alternatives and similar repositories for neural_networks_and_computation
Users that are interested in neural_networks_and_computation are comparing it to the libraries listed below
Sorting:
- β281Updated last year
- π§± Modula software packageβ237Updated last month
- Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.β173Updated 2 years ago
- Compositional Linear Algebraβ490Updated last month
- Latent Program Network (from the "Searching Latent Program Spaces" paper)β96Updated 6 months ago
- Hierarchical Associative Memory User Experienceβ103Updated 2 months ago
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.β283Updated this week
- Scalable and Stable Parallelization of Nonlinear RNNSβ22Updated 3 weeks ago
- Implementation of PSGD optimizer in JAXβ34Updated 8 months ago
- Jax/Flax rewrite of Karpathy's nanoGPTβ60Updated 2 years ago
- Named Tensors for Legible Deep Learning in JAXβ203Updated this week
- Scalable training and inference for Probabilistic Circuitsβ75Updated 2 months ago
- A modular, easy to extend GFlowNet libraryβ289Updated this week
- About A collection of AWESOME things about information geometry Topicsβ165Updated last year
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation preconditionβ¦β180Updated last week
- Minimal yet performant LLM examples in pure JAXβ158Updated last week
- Exact OU processes with JAXβ53Updated 6 months ago
- β215Updated this week
- β17Updated last year
- A package for defining deep learning models using categorical algebraic expressions.β61Updated last year
- The Energy Transformer block, in JAXβ59Updated last year
- Official implementation of Stochastic Taylor Derivative Estimator (STDE) NeurIPS2024β120Updated 9 months ago
- β56Updated 5 months ago
- Agustinus' very opiniated publication-ready plotting libraryβ69Updated 4 months ago
- seqax = sequence modeling + JAXβ167Updated last month
- Pytorch-like dataloaders for JAX.β94Updated 3 months ago
- JAX Arrays for human consumptionβ106Updated 2 months ago
- Generative Flow Networks - GFlowNetβ275Updated this week
- Turn jitted jax functions back into python source codeβ22Updated 9 months ago
- This repository contains the official code for Energy Transformer---an efficient Energy-based Transformer variant for graph classificatioβ¦β25Updated last year