watcl-lab / neural_networks_and_computationLinks
Computational abilities and efficiency of neural networks
β56Updated 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:
- β282Updated last year
- π§± Modula software packageβ282Updated last month
- Compositional Linear Algebraβ491Updated 2 months ago
- Latent Program Network (from the "Searching Latent Program Spaces" paper)β98Updated last week
- Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.β173Updated 2 years ago
- Scalable training and inference for Probabilistic Circuitsβ77Updated this week
- Agustinus' very opiniated publication-ready plotting libraryβ69Updated 5 months ago
- A modular, easy to extend GFlowNet libraryβ290Updated last week
- Scalable and Stable Parallelization of Nonlinear RNNSβ23Updated this week
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation preconditionβ¦β183Updated 2 weeks ago
- Parameter-Free Optimizers for Pytorchβ130Updated last year
- This repository contains the official code for Energy Transformer---an efficient Energy-based Transformer variant for graph classificatioβ¦β25Updated last year
- Second Order Optimization and Curvature Estimation with K-FAC in JAX.β290Updated 2 weeks ago
- Generative Flow Networks - GFlowNetβ282Updated this week
- Hierarchical Associative Memory User Experienceβ104Updated 2 months ago
- β216Updated 10 months ago
- Neural Networks and the Chomsky Hierarchyβ209Updated last year
- A MAD laboratory to improve AI architecture designs π§ͺβ129Updated 9 months ago
- Riemannian Optimization Using JAXβ53Updated last year
- IVON optimizer for neural networks based on variational learning.β72Updated 11 months ago
- Uncertainty quantification with PyTorchβ373Updated this week
- Exact OU processes with JAXβ54Updated 6 months ago
- Reliable, minimal and scalable library for pretraining foundation and world modelsβ67Updated this week
- β188Updated last month
- Pytorch-like dataloaders for JAX.β93Updated 4 months ago
- β14Updated 11 months ago
- Minimal yet performant LLM examples in pure JAXβ181Updated 2 weeks ago
- Implementation of PSGD optimizer in JAXβ33Updated 9 months ago
- About A collection of AWESOME things about information geometry Topicsβ166Updated last year
- Official implementation of Stochastic Taylor Derivative Estimator (STDE) NeurIPS2024β122Updated 10 months ago