flax

8 projetos partilham este topic do GitHub

Linhas conectam membros que estão mensuravelmente relacionados entre si. O tamanho do ponto reflete estrelas.

🧬 Membros

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

🔗 Familias relacionadas

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

Medido a partir dos tópicos do GitHub compartilhados por ambos os projetos, ponderado pela raridade de cada tópico.