flax

6 projects share this GitHub topic

Lines connect members that are measurably related to each other. Dot size reflects stars.

🧬 Members

Python code for "Probabilistic Machine learning" book by Kevin Murphy

uvadlc_notebooks

Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam…

Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning,…

Official code for Score-Based Generative Modeling through Stochastic Differential Equations (ICLR 2021, Oral)

Modular Reinforcement Learning (RL) library (implemented in PyTorch, JAX, and NVIDIA Warp) with support for…

Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint

🔗 Related families

jax 32 colab 15 controllable-generation 8 generative-models 19 gym 24 contrastive-learning 14 embedding 16

Measured from GitHub topics shared by both projects, weighted by how rare each topic is.