google-deepmind / alphatensor
☆2,670Updated 4 months ago
Related projects: ⓘ
- Flax is a neural network library for JAX that is designed for flexibility.☆5,950Updated this week
- Monte Carlo tree search in JAX☆2,312Updated last month
- JAX-based neural network library☆2,860Updated last week
- Implementation of Hinton's forward-forward (FF) algorithm - an alternative to back-propagation☆1,431Updated last year
- Train to 94% on CIFAR-10 in <6.3 seconds on a single A100. Or ~95.79% in ~110 seconds (or less!)☆1,213Updated 10 months ago
- Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.☆1,436Updated this week
- A machine learning compiler for GPUs, CPUs, and ML accelerators☆2,577Updated this week
- AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…☆4,531Updated this week
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,038Updated this week
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆8,362Updated this week
- maximal update parametrization (µP)☆1,334Updated 2 months ago
- Optax is a gradient processing and optimization library for JAX.☆1,640Updated this week
- ☆742Updated 5 months ago
- ☆2,635Updated last week
- Cramming the training of a (BERT-type) language model into limited compute.☆1,284Updated 3 months ago
- A concise but complete full-attention transformer with a set of promising experimental features from various papers☆4,573Updated last week
- ☆688Updated last year
- Model API for GALACTICA☆2,675Updated last year
- functorch is JAX-like composable function transforms for PyTorch.☆1,389Updated this week
- ManimML is a project focused on providing animations and visualizations of common machine learning concepts with the Manim Community Libr…☆2,333Updated 2 months ago
- Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.☆999Updated 2 weeks ago
- Foundation Architecture for (M)LLMs☆3,003Updated 5 months ago
- arxiv-sanity lite: tag arxiv papers of interest get recommendations of similar papers in a nice UI using SVMs over tfidf feature vectors …☆1,156Updated last year
- Supercharge Your Model Training☆5,116Updated this week
- Hopfield Networks is All You Need☆1,660Updated last year
- Training and serving large-scale neural networks with auto parallelization.☆3,045Updated 9 months ago
- 🦁 Lion, new optimizer discovered by Google Brain using genetic algorithms that is purportedly better than Adam(w), in Pytorch☆2,002Updated 3 months ago
- Structured state space sequence models☆2,361Updated 2 months ago
- PyTorch extensions for high performance and large scale training.☆3,149Updated 2 weeks ago
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆994Updated 5 months ago