Lists (1)
Stars
jaxmg: A distributed linear solver in JAX with cuSolverMg
Simplifying reinforcement learning for complex game environments
Introduction to Machine Learning Systems
Second Order Optimization and Curvature Estimation with K-FAC in JAX.
Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2023
Minimalistic 4D-parallelism distributed training framework for education purpose
A Python framework for accelerated simulation, data generation and spatial computing.
Python & Matlab code for the figures from the book "Learning Theory from First Principles" by Francis Bach
Solution Manual for Roman Vershynin's High-Dimensional Probability
A JAX-based simulator for autonomous driving research.
Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.
PyTorch compiler that accelerates training and inference. Get built-in optimizations for performance, memory, parallelism, and easily write your own.
High-performance key-value store for ML inference. 100x faster than Redis.
Gemma open-weight LLM library, from Google DeepMind
A collection of learning resources for curious software engineers
For releasing code related to compression methods for transformers, accompanying our publications
A tool to classify and statistic GPU kernel information.
Zero Bubble Pipeline Parallelism
U-Mamba: Enhancing Long-range Dependency for Biomedical Image Segmentation
DSPy: The framework for programming—not prompting—language models