search

google / paxml

Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry leading model flop utilization rates.
Apache License 2.0
456 stars 68 forks source link

Error running tutorial notebook #95

Open ajikmr opened 4 months ago

ajikmr commented 4 months ago

Hello.

I am trying to run "pax101_e2e_tutorial.ipynb' tutorial notebook.

I am getting following error:

code: task_p = tasks_lib.SingleTask.HParams(name='translation')

error: AttributeError: type object 'SingleTask' has no attribute 'HParams'

Thanks.