flax

8 proyectos comparten este topic de GitHub

Las líneas conectan a los miembros que están mediblemente relacionados entre sí. El tamaño de los puntos refleja las estrellas.

🧬 Miembros

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 de los temas de GitHub compartidos por ambos proyectos, ponderado por cuán raros son cada uno de los temas.