flax

8 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…

JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.

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

JAX library for training sub-4B foundation models for edge

🔗 Related families

jax 60 gym 39 colab 16 jupyter-notebooks 10 probabilistic-programming 13 pytorch-lightning 16 controllable-generation 8 generative-models 38

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