theano_wmf

minimal-gpt-neox-20b

fastxtend

Train fastai models faster (and other useful tools)

gaussian-splatting

Original reference implementation of "3D Gaussian Splatting for Real-Time Radiance Field Rendering"

nervana_theano

A rudimentary wrapper around the fast Maxwell kernels for GEMM and convolution operations provide...

MAE-pytorch

Unofficial PyTorch implementation of Masked Autoencoders Are Scalable Vision Learners

theano_fftconv

Convolution op for Theano based on CuFFT using scikits.cuda

matrix_fact

Matrix Factorization Library

fractional_differencing_gpu

Rapid large-scale fractional differencing with NVIDIA RAPIDS and GPU to minimize memory loss whil...

wmf

Weighted matrix factorization in Python

quantized-training

Explore training for quantized models

b3-basic-batchsize-benchmark

Experiments for the blog post "No, We Don't Have to Choose Batch Sizes As Powers Of 2"

bm25_pt

minimal pytorch implementation of bm25 (with sparse tensors)

LongMamba

Some preliminary explorations of Mamba's context scaling.

implicit

Fast Python Collaborative Filtering for Implicit Feedback Datasets

theano_optimize

Theano implementation of different optimization algorithms