mjsML
Follow
Lists (1)
Stars
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 high performance GPU simulation and graphics
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.
Make PyTorch models up to 40% faster! Thunder is a source to source compiler for PyTorch. It enables using different hardware executors at once; across one or thousands of GPUs.
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
A simple, performant and scalable Jax LLM!
Minimal and free Kubernetes distribution with Terraform
An Open-Ended Embodied Agent with Large Language Models