Machine learning and stuff
pytorch-scarf. Implementation of SCARF: Self-Supervised Contrastive Learning using Random Feature Corruption in Pytorch, a model learning a representation of tabular data using contrastive learning. It is inspired from SimCLR and uses a similar architecture and loss.
95wukong-recommendation. Implements the paper "Wukong: Towards a Scaling Law for Large-Scale Recommendation" from Meta.
87scratch-llm. From-scratch Llama 2-inspired Transformer in PyTorch for exploring tokenization, pre-training, and inference.
39evidential-deeplearning. Implementation of "Evidential Deep Learning to Quantify Classification Uncertainty" proposing a method to quantify uncertainty in a neural network.
30multitask-learning. Tensorflow implementation of three architectures for multi-task learning, a paradigm to learn different prediction tasks jointly using one model
9ctr-prediction. Implementation of some state-of-the-art deep learning architectures for CTR prediction tasks, both in Pytorch and Tensorflow
6portfolio-balance. Streamlit application to optimally re-balance a portfolio given a target allocation, current positions and market prices
5ecommerce-dash. Example of a Dash application applied to e-commerce data to represent key metrics in the form of an interactive dashboard
5final-mlp. Implements SOTA architecture for CTR predictions tasks, both in Pytorch and Tensorflow
4parqstream. High-performance Parquet dataloader for ML pipelines with multithreaded prefetching to keep GPU busy.
4rust-graph. Small graph library in rust for fun.
3fastapi-fraud-detection. Example of a fully packaged ML model in a Docker image and exposed as a REST API using FastAPI framework
2ssl-images. Implements VICReg, NT-Xent and DCL losses for contrastive self-supervised learning to generate semantically meaningful representations of images without labels.
2rust-hnsw. Rust implementation of HNSW for approximate nearest-neighbor search, generic over distance metrics and vector types.
2jax-bandits. Epsilon-greedy, UCB and Thomson sampling for Bernoulli rewards in pure jitted Jax
1rust-bandits. Real-time Rust experimentation platform with UCB and Thompson Sampling bandits, persistent state, and recovery.
1polars-plugins. Demonstrates how to implement custom expression in rust for polars and use them in python
1