google-deepmind / alphatensorLinks
☆2,753Updated last year
Alternatives and similar repositories for alphatensor
Users that are interested in alphatensor are comparing it to the libraries listed below
Sorting:
- JAX-based neural network library☆3,034Updated this week
- ☆709Updated last year
- AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…☆4,640Updated 2 months ago
- Monte Carlo tree search in JAX☆2,491Updated last month
- Optax is a gradient processing and optimization library for JAX.☆1,902Updated last week
- Flax is a neural network library for JAX that is designed for flexibility.☆6,586Updated this week
- Implementation of Hinton's forward-forward (FF) algorithm - an alternative to back-propagation☆1,481Updated last year
- ☆1,320Updated 3 weeks ago
- Model API for GALACTICA☆2,718Updated 2 years ago
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,375Updated last week
- ☆775Updated last month
- 10x faster matrix and vector operations☆2,486Updated 2 years ago
- Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.☆1,562Updated 3 weeks ago
- A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.☆2,791Updated last week
- This repository contains implementations and illustrative code to accompany DeepMind publications☆13,879Updated this week
- A pedagogical implementation of Autograd☆982Updated 5 years ago
- Library for reading and writing large multi-dimensional arrays.☆1,409Updated this week
- JAX - A curated list of resources https://github.com/google/jax☆1,820Updated 3 months ago
- Hopfield Networks is All You Need☆1,808Updated 2 years ago
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,046Updated last year
- A library of reinforcement learning components and agents☆3,691Updated this week
- Convolutions for Sequence Modeling☆883Updated 11 months ago
- A Graph Neural Network Library in Jax☆1,433Updated last year
- Training and serving large-scale neural networks with auto parallelization.☆3,134Updated last year
- A machine learning compiler for GPUs, CPUs, and ML accelerators☆3,207Updated this week
- Closed-form Continuous-time Neural Networks☆951Updated 10 months ago
- Transformers are Sample-Efficient World Models. ICLR 2023, notable top 5%.☆838Updated 7 months ago
- functorch is JAX-like composable function transforms for PyTorch.☆1,429Updated this week
- A library for differentiable nonlinear optimization☆1,898Updated 4 months ago
- Massively parallel rigidbody physics simulation on accelerator hardware.☆2,696Updated last week