nouiz

DeepLearningTutorials

Deep Learning Tutorial notes and code. See the wiki for more info.

Theano

Theano is a Python library that allows you to define, optimize, and evaluate mathematical expressions involving multi-dimensional arrays efficiently. It can use GPUs and perform efficient symbolic differentiation.

Theano-Docker

Theano Docker files

gtc2017

GTC2017 Theano Hands-on lab

tensorflow

An Open Source Machine Learning Framework for Everyone

abseil-cpp

Abseil Common Libraries (C++)

array_record

cad-capital-gains

A CLI tool built using Python for calculating your capital gains and adjusted cost base (ACB) for Canadian taxes.

conda

OS-agnostic, system-level binary package manager and ecosystem

cpython

The Python programming language

datasets

TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...

DeepLearningExamples

Deep Learning Examples

flax

Flax is a neural network library for JAX that is designed for flexibility.

Fuser

A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")

gemma

Open weights LLM from Google DeepMind.

icarogw

Icarogw is a pure python package to estimate population properties of noisy observations in presence of selection biases. The code is mostly used for gravitational waves observations.

jax

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

JAX-Toolbox

JAX-Toolbox

jax_custom_ops_and_custom_partitioning

keras

Deep Learning library for Python. Convnets, recurrent neural networks, and more. Runs on Theano or TensorFlow.

libgpuarray

Library to manipulate tensors on the GPU.

maxtext

A simple, performant and scalable Jax LLM!

praxis

pymc3

Probabilistic Programming in Python: Bayesian Modeling and Probabilistic Machine Learning with Theano

scaling-book

Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs

spack

A flexible package manager that supports multiple versions, configurations, platforms, and compilers.

t5x

TransformerEngine

A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper GPUs, to provide better performance with lower memory utilization in both training and inference.

xla

A machine learning compiler for GPUs, CPUs, and ML accelerators

XLB

XLB: Accelerated Lattice Boltzmann (XLB) based on JAX for Physics-based ML