Implementing a ChatGPT-like LLM in PyTorch from scratch, step by step

A native PyTorch Library for large model training

The AI Scientist: Towards Fully Automated Open-Ended Scientific Discovery 🧑‍🔬

Google DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Hybrid ML + physics model of the Earth's atmosphere

UNet diffusion model in pure CUDA

LLM101n: Let's build a Storyteller

Code and models accompanying "Deep Predictive Coding Networks for Video Prediction and Unsupervised Learning"

20+ high-performance LLMs with recipes to pretrain, finetune and deploy at scale.

Course to get into Large Language Models (LLMs) with roadmaps and Colab notebooks.

LLM training in simple, raw C/CUDA

Unsupervised Learning for Image Registration

Statsmodels: statistical modeling and econometrics in Python

Official inference library for Mistral models

SpikingJelly is an open-source deep learning framework for Spiking Neural Network (SNN) based on PyTorch.

The Python programming language

A library of reinforcement learning components and agents

A drop-in replacement for Scikit-Learn’s GridSearchCV / RandomizedSearchCV -- but with cutting edge hyperparameter tuning techniques.

A game theoretic approach to explain the output of any machine learning model.

Linux shell for iOS

A community effort to develop an open standard library for Medical Physics in Python. Building quality transparent software together via peer review and open source distribution. Open code is bette…

The Unix Shell

AI Toolkit for Healthcare Imaging

Tool for robust segmentation of >100 important anatomical structures in CT and MR images

This repo contains a fully developed code for Local Feature Matching using Harris Corner Detector and SIFT Descriptor

Trax — Deep Learning with Clear Code and Speed

NLTK Source