Frédéric Bastien's repositories
gemma
Open weights LLM from Google DeepMind.
xla
A machine learning compiler for GPUs, CPUs, and ML accelerators
JAX-Toolbox
JAX-Toolbox
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
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.
datasets
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...
XLB
XLB: Accelerated Lattice Boltzmann (XLB) based on JAX for Physics-based ML
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.
tensorflow
An Open Source Machine Learning Framework for Everyone
flax
Flax is a neural network library for JAX that is designed for flexibility.
DeepLearningExamples
Deep Learning Examples
Theano-Docker
Theano Docker files
libgpuarray
Library to manipulate tensors on the GPU.
conda
OS-agnostic, system-level binary package manager and ecosystem
cpython
The Python programming language
DeepLearningTutorials
Deep Learning Tutorial notes and code. See the wiki for more info.
keras
Deep Learning library for Python. Convnets, recurrent neural networks, and more. Runs on Theano or TensorFlow.
pymc3
Probabilistic Programming in Python: Bayesian Modeling and Probabilistic Machine Learning with Theano
presentations
Repo for some of my presentation
python-gsoc.github.io
Website and ideas page for Python's Google Summer of Code efforts
aiwiththebest2016
Theano presentation for http://ai.withthebest.com/