Hessian trace estimation using PyTorch and Hutch++
☆20Oct 29, 2020Updated 5 years ago
Alternatives and similar repositories for hessian_trace_estimation
Users that are interested in hessian_trace_estimation are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Gradient-based Hyperparameter Optimization Over Long Horizons☆14Sep 29, 2021Updated 4 years ago
- Official code for "Accelerating Feedforward Computation via Parallel Nonlinear Equation Solving", ICML 2021☆29Sep 25, 2021Updated 4 years ago
- ☆22Dec 3, 2021Updated 4 years ago
- Official repository for the paper: "Trees with Attention for Set Prediction Tasks" (ICML21)☆10Jan 19, 2022Updated 4 years ago
- ☆43May 3, 2024Updated last year
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- This is MPC (model predictive controller) that can predict steering and throttle to drive in a simulator.☆19Jun 13, 2017Updated 8 years ago
- Inference on non-linear dynamical systems written in JAX☆11Aug 20, 2020Updated 5 years ago
- [ICML 2024] Official Repository for the paper "Transformers Get Stable: An End-to-End Signal Propagation Theory for Language Models"☆10Jul 19, 2024Updated last year
- [NeurIPS 2020] Official Implementation: "SMYRF: Efficient Attention using Asymmetric Clustering".☆50Sep 6, 2023Updated 2 years ago
- PyTorch block-diagonal ODE CUDA solver, designed for gradient-based optimization☆16Apr 27, 2020Updated 5 years ago
- Test pytorch code with minimal computational overhead☆26Jun 8, 2023Updated 2 years ago
- Quasi-Newton Algorithm for Stochastic Optimization☆11May 20, 2022Updated 3 years ago
- This repository implements the paper "Effective Training of Convolutional Neural Networks with Low-bitwidth Weights and Activations"☆20Aug 30, 2021Updated 4 years ago
- ☆126Jun 3, 2024Updated last year
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Learning to Skip the Middle Layers of Transformers☆17Aug 7, 2025Updated 7 months ago
- A basic implementation of the paper Eigengame : PCA as a Nash Equilibrium☆21Jun 7, 2021Updated 4 years ago
- Source code for ICLR 2020 paper: "Learning to Guide Random Search"☆40Sep 2, 2024Updated last year
- The Differentiable Cross-Entropy Method☆125Aug 14, 2020Updated 5 years ago
- [NeurIPS 2020] Neural Manifold Ordinary Differential Equations (https://arxiv.org/abs/2006.10254)☆127Jul 6, 2023Updated 2 years ago
- ☆11Apr 6, 2023Updated 2 years ago
- ☆50Oct 22, 2020Updated 5 years ago
- ☆47Aug 15, 2019Updated 6 years ago
- ☆22May 16, 2022Updated 3 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click and start building anything your business needs.
- ☆21Jul 4, 2019Updated 6 years ago
- [ICLR 2025] Official PyTorch implementation of our paper for general continual learning "Advancing Prompt-Based Methods for Replay-Indepe…☆16Dec 21, 2025Updated 3 months ago
- Meta Optimal Transport☆108May 25, 2023Updated 2 years ago
- ☆15Sep 22, 2024Updated last year
- Official implementation of NeurIPS'21: Implicit SVD for Graph Representation Learning☆21Nov 4, 2021Updated 4 years ago
- ☆13Nov 26, 2023Updated 2 years ago
- ☆22Feb 4, 2025Updated last year
- ☆80Sep 4, 2021Updated 4 years ago
- Rescure useful code in TensorNetwork by google for further personal usage.☆13Dec 6, 2024Updated last year
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Performant, differentiable reinforcement learning☆125Aug 2, 2025Updated 7 months ago
- Contains code for the NeurIPS 2020 paper by Pan et al., "Continual Deep Learning by FunctionalRegularisation of Memorable Past"☆44Nov 10, 2020Updated 5 years ago
- The ✨Magical✨ JAX ML Library.☆18Jan 25, 2025Updated last year
- Public Release of Plan2vec Implementation in pyTorch☆57Oct 28, 2022Updated 3 years ago
- Source code of "Grid-to-Graph: Flexible Spatial Relational Inductive Biases for Reinforcement Learning" (AAMAS 2021).☆29Aug 19, 2021Updated 4 years ago
- ☆234Sep 3, 2023Updated 2 years ago
- Co-Adaptation of Algorithmic and Implementational Innovations in Inference-based Deep Reinforcement Learning (NeurIPS2021)☆20Oct 25, 2021Updated 4 years ago