Training PyTorch models with differential privacy
Graph Neural Network Library for PyTorch
Data manipulation and transformation for audio signal processing
Implementation of 'lightweight' GAN, proposed in ICLR 2021
ML engineer that reads papers, trains models, and ships ML models
Practice implementing softmax, attention, GPT-2 and more
Scripts for fine-tuning Meta Llama3 with composable FSDP & PEFT method
Python package built to ease deep learning on graph
A lightweight audio-to-MIDI converter with pitch bend detection
A lightweight library for PyTorch training tools and utilities
Clean, Robust, and Unified PyTorch implementation
Collection of useful data science topics along with articles
This project is a common knowledge point and code implementation
An agentic Machine Learning Engineer
Machine Learning Systems: Design and Implementation
Optax is a gradient processing and optimization library for JAX
Sacred is a tool to help you configure, andorganize IDSIA experiments
A very simple framework for state-of-the-art NLP
An open-source toolkit for BigMac-style pipeline-parallel training
Shared repository for open-sourced projects from the Google AI Lang
Reinforcement Learning for Humanoid Robot with Zero-Shot Sim2Real
SpikingJelly is an open-source deep learning framework
Numerical differential equation solvers in JAX
A comprehensive quantitative trading system with AI-powered analysis
Codes/Notebooks for AI Projects