ksetrae's starred repositories
awesome-decision-transformer
A curated list of Decision Transformer resources (continually updated)
soft-Q-learning
discrete soft Q learning(SQL) and soft Q imitation learning(SQIL) implementation in pytorch, simple!
imitation-learning
Imitation learning algorithms
DirectML
DirectML is a high-performance, hardware-accelerated DirectX 12 library for machine learning. DirectML provides GPU acceleration for common machine learning tasks across a broad range of supported hardware and drivers, including all DirectX 12-capable GPUs from vendors such as AMD, Intel, NVIDIA, and Qualcomm.
PyTorch-LBFGS
A PyTorch implementation of L-BFGS.
awesome-point-cloud-deep-learning
Paper list of deep learning on point clouds.
MBM-Colorization
Image colorization with a Multivariate Bernoulli Mixture Density network.
sparse_learning
Sparse learning library and sparse momentum resources.
modern-srwm
Official repository for the paper "A Modern Self-Referential Weight Matrix That Learns to Modify Itself" (ICML 2022 & NeurIPS 2021 Deep RL Workshop) and "Accelerating Neural Self-Improvement via Bootstrapping" (ICLR 2023 Workshop)
pytorch-maml
An Implementation of Model-Agnostic Meta-Learning in PyTorch with Torchmeta
pytorch-meta
A collection of extensions and data-loaders for few-shot learning & meta-learning in PyTorch
fast-soft-sort
Fast Differentiable Sorting and Ranking
torchswarm
A fast implementation of Particle Swarm Optimization using PyTorch in GPU
SimulatedAnnealing
Pytorch Optimizer for Simulated Annealing
SpykeTorch
High-speed simulator of convolutional spiking neural networks with at most one spike per neuron.
pytorch-optimizer
torch-optimizer -- collection of optimizers for Pytorch