65 packages found

chex

Chex: Testing made fun, in JAX!
  1. jax
  2. testing
  3. debugging
  4. python
  5. machine
  6. learning
36 Contributors
0.1.89published 1 month agoApache-2.0

pytorch-transformers-pvt-nightly

Repository of pre-trained NLP Transformer models: BERT & RoBERTa, GPT & GPT-2, Transformer-XL, XLNet and XLM
  1. NLP
  2. deep
  3. learning
  4. transformer
  5. pytorch
  6. BERT
  7. GPT
  8. GPT-2
  9. google
  10. openai
  11. CMU
  12. deep-learning
  13. flax
  14. hacktoberfest
  15. jax
  16. language-model
  17. language-models
  18. machine-learning
  19. model-hub
  20. natural-language-processing
  21. nlp-library
  22. pretrained-models
  23. python
  24. pytorch-transformers
  25. seq2seq
  26. speech-recognition
  27. tensorflow
721 Contributors
1.2.0.dev201909261000published 6 years agoApache-2.0

mctx

Monte Carlo tree search in JAX.
  1. jax
  2. planning
  3. reinforcement-learning
  4. python
  5. machine
  6. learning
  7. monte-carlo-tree-search
8 Contributors
0.0.5published 1 year agoApache-2.0

diffrax

GPU+autodiff-capable ODE/SDE/CDE solvers written in JAX.
  1. deep-learning
  2. differential-equations
  3. diffrax
  4. dynamical-systems
  5. equinox
  6. jax
  7. neural-differential-equations
  8. machine-learning
  9. neural-networks
36 Contributors
0.7.0published 4 weeks agoOther

skscope

Sparsity-Constraint OPtimization via itErative-algorithm
  1. auto-differentiation
  2. jax
  3. non-convex-optimization
  4. nonlinear-optimization
  5. python
  6. scikit-learn
  7. sparsity-optimization
0.1.8published 10 months agoMIT

wandb

A CLI and library for interacting with the Weights & Biases API.
  1. ai
  2. collaboration
  3. data-science
  4. data-versioning
  5. deep-learning
  6. experiment-track
  7. hyperparameter-optimization
  8. hyperparameter-search
  9. hyperparameter-tuning
  10. jax
  11. keras
  12. machine-learning
  13. ml-platform
  14. mlops
  15. model-versioning
  16. pytorch
  17. reinforcement-learning
  18. reproducibility
  19. tensorflow
196 Contributors
0.19.9published 1 week agoOther

jax

Differentiate, compile, and transform Numpy code.
  1. jax
568 Contributors
0.5.3published 3 weeks agoApache-2.0

transformers

State-of-the-art Machine Learning for JAX, PyTorch and TensorFlow
  1. NLP
  2. vision
  3. speech
  4. deep
  5. learning
  6. transformer
  7. pytorch
  8. tensorflow
  9. jax
  10. BERT
  11. GPT-2
  12. Wav2Vec2
  13. ViT
  14. deep-learning
  15. flax
  16. hacktoberfest
  17. language-model
  18. language-models
  19. machine-learning
  20. model-hub
  21. natural-language-processing
  22. nlp-library
  23. pretrained-models
  24. python
  25. pytorch-transformers
  26. seq2seq
  27. speech-recognition
726 Contributors
4.51.0published 3 days agoApache-2.0

tensorly

Tensor learning in Python.
  1. cupy
  2. decomposition
  3. jax
  4. machine-learning
  5. mxnet
  6. numpy
  7. python
  8. pytorch
  9. regression
  10. tensor
  11. tensor-algebra
  12. tensor-decomposition
  13. tensor-factorization
  14. tensor-learning
  15. tensor-methods
  16. tensor-regression
  17. tensorflow
  18. tensorly
58 Contributors
0.9.0published 5 months agoBSD-2-Clause

pgx

♟️ Vectorized RL game environments in JAX
  1. ai
  2. alphazero
  3. artificial-intelligence
  4. backgammon
  5. bridge-game
  6. chess
  7. game
  8. go-game
  9. jax
  10. machine-learning
  11. poker
  12. python
  13. reinforcement-learning
  14. shogi
17 Contributors
2.6.0published 1 month agoApache-2.0
Showing 1 to 10 of 65 results