Ecosyste.ms: Packages

An open API service providing package, version and dependency metadata of many open source software ecosystems and registries.

pypi.org "flax" keyword

hal9 2.0.9
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
30 versions - Latest release: about 8 hours ago - 1.21 thousand downloads last month - 22,366 stars on GitHub - 2 maintainers
Top 0.1% on pypi.org
transformers 4.40.2
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow
141 versions - Latest release: 10 days ago - 2,589 dependent packages - 31,800 dependent repositories - 27.6 million downloads last month - 126,076 stars on GitHub - 5 maintainers
easydel 0.0.65
An open-source library to make training faster and more optimized in Jax/Flax
45 versions - Latest release: about 13 hours ago - 797 downloads last month - 66 stars on GitHub - 1 maintainer
jestimator 0.3.3
Implementation of the Amos optimizer from the JEstimator lib.
5 versions - Latest release: over 1 year ago - 4 dependent repositories - 50 downloads last month - 77 stars on GitHub - 1 maintainer
fjformer 0.0.57
Embark on a journey of paralleled/unparalleled computational prowess with FJFormer - an arsenal o...
46 versions - Latest release: 2 days ago - 1 dependent package - 1.15 thousand downloads last month - 21 stars on GitHub - 1 maintainer
grouped-query-attention-pytorch 0.3.0
grouped-query-attention-pytorch
4 versions - Latest release: 7 days ago - 420 downloads last month - 126,076 stars on GitHub - 1 maintainer
transformers-machinify 4.27.0.dev0
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow
1 version - Latest release: about 1 year ago - 1 dependent package - 40 downloads last month - 126,076 stars on GitHub - 2 maintainers
axlearn 0.0.1
AXLearn
3 versions - Latest release: 9 months ago - 137 downloads last month - 126,076 stars on GitHub - 1 maintainer
transformers-phobert 3.1.2
State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch. Note that the tokeni...
2 versions - Latest release: over 3 years ago - 1 dependent repositories - 26 downloads last month - 126,076 stars on GitHub - 1 maintainer
in-transformers 1.0.0
State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow
1 version - Latest release: over 1 year ago - 20 downloads last month - 125,970 stars on GitHub - 1 maintainer
hf-argparser 0.1.1
Standalone version of the HuggingFace ArgParser from the Transformers library
2 versions - Latest release: 7 months ago - 1 dependent repositories - 20 downloads last month - 125,970 stars on GitHub - 1 maintainer
Top 9.6% on pypi.org
tevatron 0.1.0
Tevatron: A toolkit for learning and running deep dense retrieval models.
1 version - Latest release: about 2 years ago - 1 dependent package - 1 dependent repositories - 303 downloads last month - 373 stars on GitHub - 1 maintainer
Top 0.7% on pypi.org
pytorch-transformers 1.2.0
Repository of pre-trained NLP Transformer models: BERT & RoBERTa, GPT & GPT-2, Transformer-XL, XL...
4 versions - Latest release: over 4 years ago - 16 dependent packages - 772 dependent repositories - 46.6 thousand downloads last month - 123,549 stars on GitHub - 1 maintainer
Top 0.6% on pypi.org
pytorch-pretrained-bert 0.6.2
PyTorch version of Google AI BERT model with script to load Google pre-trained models
10 versions - Latest release: about 5 years ago - 11 dependent packages - 940 dependent repositories - 25.4 thousand downloads last month - 125,524 stars on GitHub - 1 maintainer
Top 1.3% on pypi.org
orbax-checkpoint 0.5.11
Orbax Checkpoint
41 versions - Latest release: 6 days ago - 16 dependent packages - 313 dependent repositories - 21.2 million downloads last month - 221 stars on GitHub - 1 maintainer
helx-base 1.0.0
Interoperate among reinforcement learning libraries with jax, pytorch, gym and dm_env
1 version - Latest release: 5 months ago - 3 dependent packages - 11 downloads last month - 10 stars on GitHub - 1 maintainer
Top 2.7% on pypi.org
clu 0.0.12
Set of libraries for ML training loops in JAX.
14 versions - Latest release: about 1 month ago - 14 dependent packages - 350 dependent repositories - 2.52 million downloads last month - 295 stars on GitHub - 3 maintainers
orbax-export 0.0.4
Orbax Export
5 versions - Latest release: 4 months ago - 1 dependent package - 1 dependent repositories - 599 downloads last month - 221 stars on GitHub - 1 maintainer
hidiffusion 0.1.8
HiDiffusion: A training-free method to increase the resolution and speed of diffusion models.
6 versions - Latest release: 9 days ago - 2.07 thousand downloads last month - 22,366 stars on GitHub - 1 maintainer
nanodl 1.2.4.dev1 💰
A Jax-based library for designing and training transformer models from scratch.
6 versions - Latest release: 4 days ago - 110 downloads last month - 256 stars on GitHub - 1 maintainer
pixelyai-serve 0.0.8 removed
serve utilises of pixelyai in jax and torch
9 versions - Latest release: 6 months ago - 245 downloads last month - 63 stars on GitHub - 1 maintainer
explainax 0.0.1
JAX-based Model Explanation and Interpretation Library
1 version - Latest release: about 1 year ago - 15 downloads last month - 0 stars on GitHub - 1 maintainer
redco 0.4.16
NAACL '24 (Demo) / MlSys @ NeurIPS '23 - RedCoast: A Lightweight Tool to Automate Distributed Tra...
17 versions - Latest release: 27 days ago - 18 downloads last month - 40 stars on GitHub - 1 maintainer
chemise 0.0.1a1
Wrapper for training flax models
2 versions - Latest release: about 1 year ago - 30 downloads last month - 1 stars on GitHub - 1 maintainer
xpag 0.2.9
xpag: Exploring Agents
18 versions - Latest release: 6 months ago - 82 downloads last month - 20 stars on GitHub - 1 maintainer
helx 1.1.5
Interoperate among reinforcement learning libraries with jax, pytorch, gym and dm_env
12 versions - Latest release: 4 months ago - 115 downloads last month - 10 stars on GitHub - 1 maintainer
aws-fortuna 10.7.8
A Library for Uncertainty Quantification.
44 versions - Latest release: over 1 year ago - 2 dependent repositories - 602 downloads last month - 853 stars on GitHub - 2 maintainers
jaxfss 0.1.0
JAX/Flax implementation of finite-size scaling
2 versions - Latest release: over 1 year ago - 18 downloads last month - 8 stars on GitHub - 1 maintainer
gradcache 0.0.1
Run Effective Large Batch Contrastive Learning Beyond GPU/TPU Memory Constraint
2 versions - Latest release: almost 2 years ago - 1 dependent repositories - 13 downloads last month - 293 stars on GitHub - 1 maintainer
hyper-nn 0.2.2
Easy hypernetworks in Pytorch and Flax
6 versions - Latest release: over 1 year ago - 1 dependent repositories - 85 downloads last month - 90 stars on GitHub - 1 maintainer
vit-flax 0.0.3
Vision Transformer, ViT - Flax
3 versions - Latest release: over 3 years ago - 1 dependent repositories - 21 downloads last month - 13 stars on GitHub - 1 maintainer
Top 8.8% on pypi.org
pytorch-transformers-pvt-nightly 1.2.0.dev201909261000
Repository of pre-trained NLP Transformer models: BERT & RoBERTa, GPT & GPT-2, Transformer-XL, XL...
966 versions - Latest release: over 4 years ago - 1 dependent repositories - 8.84 thousand downloads last month - 125,840 stars on GitHub - 1 maintainer
python-flax-id 1.1.0
Python implementation of Flax ID algorithm
2 versions - Latest release: about 6 years ago - 1 dependent repositories - 22 downloads last month - 1 maintainer
perceiver-jax 0.0.3
Perceiver - JAX
3 versions - Latest release: about 3 years ago - 1 dependent repositories - 22 downloads last month - 10 stars on GitHub - 1 maintainer
jimm 0.0.1
JAX Image Models
1 version - Latest release: over 3 years ago - 1 dependent repositories - 19 downloads last month - 1 maintainer
jax-resnet 0.0.4
Framework-agnostic library for checking array shapes at runtime.
2 versions - Latest release: almost 3 years ago - 1 dependent package - 2 dependent repositories - 53 downloads last month - 101 stars on GitHub - 1 maintainer
jax-models 0.0.5
Unofficial JAX implementations of deep learning research papers
5 versions - Latest release: about 2 years ago - 1 dependent repositories - 64 downloads last month - 139 stars on GitHub - 1 maintainer
jax-fid 0.0.2
FID computation in Jax/Flax.
2 versions - Latest release: over 2 years ago - 1 dependent repositories - 15 downloads last month - 21 stars on GitHub - 1 maintainer
graphmlp-flax 0.0.1 removed
Minimal Flax Implementation of Graph-MLP: Node Classification without Message Passing in Graph
1 version - Latest release: almost 3 years ago - 1 stars on GitHub - 1 maintainer
gmlp-flax 0.0.1 removed
Flax implementation of gMLP from Pay Attention to MLPs
1 version - Latest release: almost 3 years ago - 2 stars on GitHub - 1 maintainer
fnet-flax 0.0.1 removed
Minimal Flax implementation of FNet: Mixing Tokens with Fourier Transforms
1 version - Latest release: almost 3 years ago - 10 downloads last month - 0 stars on GitHub - 1 maintainer
Top 8.3% on pypi.org
flaxmodels 0.1.3
A collection of pretrained models in Flax.
4 versions - Latest release: 9 months ago - 4 dependent repositories - 141 thousand downloads last month - 222 stars on GitHub - 1 maintainer
flax-extra 0.0.2
The package provides extra flexibility to Flax using ideas originated at Trax
3 versions - Latest release: over 2 years ago - 1 dependent repositories - 33 downloads last month - 1 stars on GitHub - 1 maintainer
ramsey 0.2.1
Probabilistic deep learning using JAX
5 versions - Latest release: 7 months ago - 1 dependent repositories - 46 downloads last month - 10 stars on GitHub - 1 maintainer
hypernn 0.1.1 removed
Easy hypernetworks in Pytorch and Flax
2 versions - Latest release: almost 2 years ago - 48 stars on GitHub
safejax 0.5.0 💰
Serialize JAX, Flax, Haiku, or Objax model params with `safetensors`
6 versions - Latest release: over 1 year ago - 81 downloads last month - 38 stars on GitHub - 1 maintainer
hrr 1.2.3
Holographic Reduced Representations
5 versions - Latest release: about 1 year ago - 36 downloads last month - 17 stars on GitHub - 1 maintainer
rationalnets 0.1.0
JAX/Flax implementation of rational neural nets
2 versions - Latest release: over 1 year ago - 1 dependent repositories - 32 downloads last month - 1 stars on GitHub - 1 maintainer
ymdl 0.2.6
A deep learning package manager.
5 versions - Latest release: almost 2 years ago - 1 dependent repositories - 21 downloads last month - 9 stars on GitHub - 1 maintainer
flaim 0.0.5
Flax Image Models
5 versions - Latest release: over 1 year ago - 54 downloads last month - 12 stars on GitHub - 1 maintainer
bobbin 0.0.2
Tools for making training loops with flax.linen models.
2 versions - Latest release: about 1 year ago - 18 downloads last month - 3 stars on GitHub - 1 maintainer
soujpg-diffusers 0.27.2
State-of-the-art diffusion in PyTorch and JAX.
4 versions - Latest release: about 2 months ago - 72 downloads last month - 22,366 stars on GitHub - 1 maintainer
resmlp-flax 0.0.1 removed
Minimal Flax implementation of ResMLP: Feedforward networks for image classification with data-ef...
1 version - Latest release: almost 3 years ago - 12 downloads last month - 1 stars on GitHub
mlpmixer-flax 0.0.1 removed
Flax implementation of MLP-Mixer from MLP-Mixer: An all-MLP Architecture for Vision
1 version - Latest release: almost 3 years ago - 8 stars on GitHub - 1 maintainer